Skip to content

Commit 312602f

Browse files
committed
fix shard_specific_patterns
1 parent 4ece734 commit 312602f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exo/download/hf/hf_helpers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -409,7 +409,7 @@ def get_allow_patterns(weight_map: Dict[str, str], shard: Shard) -> List[str]:
409409
elif shard.is_last_layer():
410410
shard_specific_patterns.add(sorted_file_names[-1])
411411
else:
412-
shard_specific_patterns = set("*.safetensors")
412+
shard_specific_patterns = set(["*.safetensors"])
413413
if DEBUG >= 2: print(f"get_allow_patterns {weight_map=} {shard=} {shard_specific_patterns=}")
414414
return list(default_patterns | shard_specific_patterns)
415415

0 commit comments

Comments
 (0)