From 800ccbf22a9940f46e31f4d70fa6d23194a43100 Mon Sep 17 00:00:00 2001 From: Irene Dea Date: Mon, 22 Jan 2024 16:50:15 -0800 Subject: [PATCH] fix --- scripts/misc/download_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/misc/download_model.py b/scripts/misc/download_model.py index 79f7576b5c..a81096084a 100644 --- a/scripts/misc/download_model.py +++ b/scripts/misc/download_model.py @@ -115,7 +115,8 @@ def parse_args() -> argparse.Namespace: download_from_hf_hub(args.model, save_dir=args.save_dir, token=args.token, + tokenizers_only=args.tokenizer_only, prefer_safetensors=args.prefer_safetensors) elif download_from == 'oras': download_from_oras(args.model, args.config_file, args.credentials_dir, - args.save_dir, args.concurrency) + args.save_dir, args.tokenizer_only, args.concurrency)