Browse Source

fix allow patterns

Baye Dieng 1 năm trước cách đây
mục cha
commit
d7fff0d62e
1 tập tin đã thay đổi với 3 bổ sung1 xóa
  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:
     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