From 849a64811a795489ded43da4ac0aeff7d1c8a9ed Mon Sep 17 00:00:00 2001 From: litagin02 Date: Wed, 7 Feb 2024 10:44:17 +0900 Subject: [PATCH] Refactor: move code to main block --- initialize.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/initialize.py b/initialize.py index 6dfd03d..c163ef9 100644 --- a/initialize.py +++ b/initialize.py @@ -90,17 +90,18 @@ def download_jvnv_models(): ) -parser = argparse.ArgumentParser() -parser.add_argument("--skip_jvnv", action="store_true") -args = parser.parse_args() +if __name__ == "__main__": + 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() -download_pretrained_models() + download_pretrained_models() -download_jp_extra_pretrained_models() + download_jp_extra_pretrained_models() -if not args.skip_jvnv: - download_jvnv_models() + if not args.skip_jvnv: + download_jvnv_models()