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

Refactor: move code to main block

Browse Source
This commit is contained in:
litagin02
2024-02-07 10:44:17 +09:00
parent 27518345dd
commit 849a64811a
1 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
initialize.py
View File

@@ -90,17 +90,18 @@ def download_jvnv_models():
) )
parser = argparse.ArgumentParser() if __name__ == "__main__":
parser.add_argument("--skip_jvnv", action="store_true") parser = argparse.ArgumentParser()
args = parser.parse_args() parser.add_argument("--skip_jvnv", action="store_true")
args = parser.parse_args()
download_bert_models() download_bert_models()
download_slm_model() download_slm_model()
download_pretrained_models() download_pretrained_models()
download_jp_extra_pretrained_models() download_jp_extra_pretrained_models()
if not args.skip_jvnv: if not args.skip_jvnv:
download_jvnv_models() download_jvnv_models()