Add skip jvnv model download (useful for colab etc)
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
import argparse
|
||||
import json
|
||||
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_slm_model()
|
||||
@@ -97,4 +102,5 @@ download_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