fix allow patterns

d7fff0d62ea2797f5fc87ae05048374b2cc9c3f6 · 2024-09-21 14:51:55 +0000 · Baye Dieng

Files touched

M exo/download/hf/hf_helpers.py

Diff

commit d7fff0d62ea2797f5fc87ae05048374b2cc9c3f6
Author: Baye Dieng <bayedieng98@gmail.com>
Date:   Sat Sep 21 14:51:55 2024 +0000

    fix allow patterns
---
 exo/download/hf/hf_helpers.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/exo/download/hf/hf_helpers.py b/exo/download/hf/hf_helpers.py
index 3197605d..260f0f7f 100644
--- a/exo/download/hf/hf_helpers.py
+++ b/exo/download/hf/hf_helpers.py
@@ -406,4 +406,6 @@ def get_allow_patterns(weight_map: Dict[str, str], shard: Shard) -> List[str]:
   else:
     shard_specific_patterns = ["*.safetensors"]
   if DEBUG >= 2: print(f"get_allow_patterns {weight_map=} {shard=} {shard_specific_patterns=}")
-  return list(default_patterns | shard_specific_patterns)
+  allowed_patterns = list(default_patterns)
+  allowed_patterns.extend(shard_specific_patterns)
+  return allowed_patterns

← cb633868 update readme with tips for performance for apple silicon ma · back to Exo · implement tailscale discovery module 93224799 →