Skip to content

Commit

Permalink
download hugging face model
Browse files Browse the repository at this point in the history
  • Loading branch information
bayedieng committed Sep 9, 2024
1 parent fb4ddf3 commit 12ae92c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion exo/download/hf/hf_helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -399,5 +399,5 @@ def get_allow_patterns(weight_map: Dict[str, str], shard: Shard) -> List[str]:
elif shard.is_last_layer():
shard_specific_patterns.append(sorted_file_names[-1])
else:
shard_specific_patterns = ["*.safetensors"]
shard_specific_patterns = ["*.safetensors", "*Q4_K_M*.gguf"]
return list(set(default_patterns + shard_specific_patterns)) # Remove duplicates

0 comments on commit 12ae92c

Please sign in to comment.