Browse Source

fix allow patterns

Baye Dieng 10 months ago
parent
commit
d7fff0d62e
1 changed files with 3 additions and 1 deletions
  1. 3 1
      exo/download/hf/hf_helpers.py

+ 3 - 1
exo/download/hf/hf_helpers.py

@@ -406,4 +406,6 @@ def get_allow_patterns(weight_map: Dict[str, str], shard: Shard) -> List[str]:
   else:
   else:
     shard_specific_patterns = ["*.safetensors"]
     shard_specific_patterns = ["*.safetensors"]
   if DEBUG >= 2: print(f"get_allow_patterns {weight_map=} {shard=} {shard_specific_patterns=}")
   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