Fix ignore patterns

7ba31254 · Casper Hansen · db8ba322 · 7ba31254
Commit 7ba31254 authored Sep 13, 2023 by Casper Hansen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

awq/models/base.py awq/models/base.py +3 -3

No files found.
--- a/awq/models/base.py
+++ b/awq/models/base.py
@@ -292,10 +292,10 @@ class BaseAWQForCausalLM(nn.Module):
        if not os.path.isdir(model_path):
            ignore_patterns = ["*msgpack*", "*h5*"]
            if safetensors:
-                ignore_patterns.extend(["*.pt", "*.bin"])
+                ignore_patterns.extend(["*.pt*", "*.bin*"])
            else:
-                ignore_patterns.append("*safetensors*")
-
+                ignore_patterns.append("*.safetensors*")
+            
            model_path = snapshot_download(model_path, ignore_patterns=ignore_patterns)
        
        model_weights_path = model_path + f'/{model_filename}'