瀏覽代碼

fix allow patterns

Baye Dieng 1 年之前
父節點
當前提交
d7fff0d62e
共有 1 個文件被更改,包括 3 次插入1 次删除
  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