feat: Enable recursive search of HFFS.ls when using from_pretrained (…

…#1656) Enable rercursive ls for hffs. Otherwise models in subfolder wouldn't be found. Co-authored-by: Andrei <[email protected]>
abetlen · Aug 7, 2024 · 5e39a85 · 5e39a85
1 parent 131db40
commit 5e39a85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py
@@ -2160,7 +2160,7 @@ def from_pretrained(
 
         files = [
             file["name"] if isinstance(file, dict) else file
-            for file in hffs.ls(repo_id)
+            for file in hffs.ls(repo_id, recursive=True))
         ]
 
         # split each file into repo_id, subfolder, filename