Răsfoiți Sursa

fix shard_specific_patterns

Alex Cheema 5 luni în urmă
părinte
comite
312602fa13
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  1. 1 1
      exo/download/hf/hf_helpers.py

+ 1 - 1
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)