fix

mosaicml · Jan 23, 2024 · 800ccbf · 800ccbf
1 parent f7f5846
commit 800ccbf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)