fix code quality

mosaicml · Jan 23, 2024 · 854c31c · 854c31c
1 parent de1601c
commit 854c31c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/llmfoundry/utils/model_download_utils.py b/llmfoundry/utils/model_download_utils.py
@@ -51,7 +51,7 @@ def download_from_hf_hub(
     model: str,
     save_dir: str,
     prefer_safetensors: bool = True,
-    tokenizers_only: bool = False,
+    tokenizer_only: bool = False,
     token: Optional[str] = None,
 ):
     """Downloads model files from a Hugging Face Hub model repo.
@@ -102,7 +102,7 @@ def download_from_hf_hub(
             f'No supported model weights found in repo {model}.' +
             ' Please make sure the repo contains either safetensors or pytorch weights.'
         )
-    
+
     allow_patterns = TOKENIZER_FILES if tokenizer_only else None
 
     download_start = time.time()

diff --git a/scripts/misc/download_model.py b/scripts/misc/download_model.py
@@ -119,5 +119,9 @@ def parse_args() -> argparse.Namespace:
                              tokenizer_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, tokenizers_only=args.tokenizer_only, args.concurrency)
+        download_from_oras(args.model,
+                           args.config_file,
+                           args.credentials_dir,
+                           args.save_dir,
+                           tokenizer_only=args.tokenizer_only,
+                           concurrency=args.concurrency)