Skip to content

Commit 5d4758f

Browse files
committed
fix: additional_files support glob patterns
1 parent e712cff commit 5d4758f

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

llama_cpp/llama.py

+8-4
Original file line numberDiff line numberDiff line change
@@ -2254,7 +2254,7 @@ def from_pretrained(
22542254
Returns:
22552255
A Llama model."""
22562256
try:
2257-
from huggingface_hub import hf_hub_download, HfFileSystem
2257+
from huggingface_hub import hf_hub_download, snapshot_download, HfFileSystem
22582258
from huggingface_hub.utils import validate_repo_id
22592259
except ImportError:
22602260
raise ImportError(
@@ -2319,10 +2319,14 @@ def from_pretrained(
23192319
)
23202320

23212321
if len(matching_additional_files) > 1:
2322-
raise ValueError(
2323-
f"Multiple files found in {repo_id} matching {additonal_file_name}\n\n"
2324-
f"Available Files:\n{json.dumps(files)}"
2322+
snapshot_download(
2323+
repo_id=repo_id,
2324+
allow_patterns=additonal_file_name,
2325+
local_dir=local_dir,
2326+
local_dir_use_symlinks=local_dir_use_symlinks,
2327+
cache_dir=cache_dir,
23252328
)
2329+
continue
23262330

23272331
(matching_additional_file,) = matching_additional_files
23282332

0 commit comments

Comments
 (0)