fix shard_specific_patterns

312602fa13f89a2199c79154996109679dbebebd · 2024-11-19 15:52:21 +0400 · Alex Cheema

Files touched

M exo/download/hf/hf_helpers.py

Diff

commit 312602fa13f89a2199c79154996109679dbebebd
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Tue Nov 19 15:52:21 2024 +0400

    fix shard_specific_patterns
---
 exo/download/hf/hf_helpers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exo/download/hf/hf_helpers.py b/exo/download/hf/hf_helpers.py
index 4729e5f6..37d4d04b 100644
--- a/exo/download/hf/hf_helpers.py
+++ b/exo/download/hf/hf_helpers.py
@@ -409,7 +409,7 @@ def get_allow_patterns(weight_map: Dict[str, str], shard: Shard) -> List[str]:
     elif shard.is_last_layer():
       shard_specific_patterns.add(sorted_file_names[-1])
   else:
-    shard_specific_patterns = set("*.safetensors")
+    shard_specific_patterns = set(["*.safetensors"])
   if DEBUG >= 2: print(f"get_allow_patterns {weight_map=} {shard=} {shard_specific_patterns=}")
   return list(default_patterns | shard_specific_patterns)

← 4ece7342 always run tinygrad stuff on same thread. tricky because of · back to Exo · add --default-model command line arg 0ab302a3 →