Add skip jvnv model download (useful for colab etc)

This commit is contained in:
litagin02
2024-02-04 18:33:09 +09:00
parent 079b5f5ee3
commit 48e24c020b

View File

@@ -1,3 +1,4 @@
import argparse
import json import json
from pathlib import Path from pathlib import Path
@@ -89,6 +90,10 @@ def download_jvnv_models():
) )
parser = argparse.ArgumentParser()
parser.add_argument("--skip_jvnv", action="store_true")
args = parser.parse_args()
download_bert_models() download_bert_models()
download_slm_model() download_slm_model()
@@ -97,4 +102,5 @@ download_pretrained_models()
download_jp_extra_pretrained_models() download_jp_extra_pretrained_models()
download_jvnv_models() if not args.skip_jvnv:
download_jvnv_models()