tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fix hf push

Browse Source
This commit is contained in:
litagin02
2024-02-17 11:31:35 +09:00
parent ff730476e7
commit 8b76109b75
2 changed files with 43 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
train_ms_jp_extra.py
View File

@@ -574,15 +574,18 @@ def run():
),
for_infer=True,
)
if hps.repo_id is not None:
api.upload_folder(
repo_id=hps.repo_id,
folder_path=model_dir,
)
api.upload_folder(
repo_id=hps.repo_id,
folder_path=config.out_dir,
)
if hps.repo_id is not None:
api.upload_folder(
repo_id=hps.repo_id,
folder_path=model_dir,
path_in_repo=f"Data/{config.model_name}/models",
delete_patterns="*.pth",
)
api.upload_folder(
repo_id=hps.repo_id,
folder_path=config.out_dir,
path_in_repo=f"model_assets/{config.model_name}",
)
if pbar is not None:
pbar.close()
@@ -937,11 +940,14 @@ def train_and_evaluate(
if hps.repo_id is not None:
api.upload_folder(
repo_id=hps.repo_id,
folder_path=model_dir,
folder_path=hps.model_dir,
path_in_repo=f"Data/{config.model_name}/models",
delete_patterns="*.pth",
)
api.upload_folder(
repo_id=hps.repo_id,
folder_path=config.out_dir,
path_in_repo=f"model_assets/{config.model_name}",
)
global_step += 1