Skip to content

Commit 0084f10

Browse files
committed
fix: additional_files support glob patterns
1 parent 7c4aead commit 0084f10

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
@@ -2255,7 +2255,7 @@ def from_pretrained(
22552255
Returns:
22562256
A Llama model."""
22572257
try:
2258-
from huggingface_hub import hf_hub_download, HfFileSystem
2258+
from huggingface_hub import hf_hub_download, snapshot_download, HfFileSystem
22592259
from huggingface_hub.utils import validate_repo_id
22602260
except ImportError:
22612261
raise ImportError(
@@ -2320,10 +2320,14 @@ def from_pretrained(
23202320
)
23212321

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

23282332
(matching_additional_file,) = matching_additional_files
23292333

0 commit comments

Comments
 (0)