Add skip jvnv model download (useful for colab etc)
This commit is contained in:
@@ -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()
|
||||||
|
|||||||
Reference in New Issue
Block a user