diff --git a/initialize.py b/initialize.py index e47ab68..f30846b 100644 --- a/initialize.py +++ b/initialize.py @@ -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()