Skip to content

GitLab

Commit 7ba31254 authored by Casper Hansen's avatar Casper Hansen
Browse files

Fix ignore patterns

parent db8ba322
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
awq/models/base.py
View file @ 7ba31254
...@@ -292,10 +292,10 @@ class BaseAWQForCausalLM(nn.Module): ...@@ -292,10 +292,10 @@ class BaseAWQForCausalLM(nn.Module):
if not os.path.isdir(model_path): if not os.path.isdir(model_path):
ignore_patterns = ["*msgpack*", "*h5*"] ignore_patterns = ["*msgpack*", "*h5*"]
if safetensors: if safetensors:
ignore_patterns.extend(["*.pt", "*.bin"]) ignore_patterns.extend(["*.pt*", "*.bin*"])
else: else:
ignore_patterns.append("*safetensors*") ignore_patterns.append("*.safetensors*")
model_path = snapshot_download(model_path, ignore_patterns=ignore_patterns) model_path = snapshot_download(model_path, ignore_patterns=ignore_patterns)
model_weights_path = model_path + f'/{model_filename}' model_weights_path = model_path + f'/{model_filename}'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment