fix: additional_files support glob patterns

abetlen · Nov 1, 2024 · 5d4758f · 5d4758f
1 parent e712cff
commit 5d4758f
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py
@@ -2254,7 +2254,7 @@ def from_pretrained(
         Returns:
             A Llama model."""
         try:
-            from huggingface_hub import hf_hub_download, HfFileSystem
+            from huggingface_hub import hf_hub_download, snapshot_download, HfFileSystem
             from huggingface_hub.utils import validate_repo_id
         except ImportError:
             raise ImportError(
@@ -2319,10 +2319,14 @@ def from_pretrained(
                     )
 
                 if len(matching_additional_files) > 1:
-                    raise ValueError(
-                        f"Multiple files found in {repo_id} matching {additonal_file_name}\n\n"
-                        f"Available Files:\n{json.dumps(files)}"
+                    snapshot_download(
+                        repo_id=repo_id,
+                        allow_patterns=additonal_file_name,
+                        local_dir=local_dir,
+                        local_dir_use_symlinks=local_dir_use_symlinks,
+                        cache_dir=cache_dir,
                     )
+                    continue
 
                 (matching_additional_file,) = matching_additional_files