ignore onnx in ignore_patterns (#570)

casper-hansen · Aug 4, 2024 · 4d5c4a9 · 4d5c4a9
1 parent b9e9b73
commit 4d5c4a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/awq/models/base.py b/awq/models/base.py
@@ -573,7 +573,7 @@ def _load_config(
     ):
         # [STEP 1] Download model if path is not a directory
         if not os.path.isdir(model_path):
-            ignore_patterns = ["*msgpack*", "*h5*", "optimizer.pt"]
+            ignore_patterns = ["*msgpack*", "*h5*", "optimizer.pt", "*.onnx*"]
             if safetensors:
                 ignore_patterns.extend(["*.pt*", "*.bin*", "consolidated*"])
             else: