From 5d4758f4299cbbb039c17a362d3562ab933d5253 Mon Sep 17 00:00:00 2001
From: xianml <xian@bentoml.com>
Date: Wed, 16 Oct 2024 12:08:26 +0800
Subject: [PATCH] fix: additional_files support glob patterns

---
 llama_cpp/llama.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py
index d15a88b00..f9e47132e 100644
--- 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