From 4d5c4a9e36ae053edab68f10998e4266b45e6a8f Mon Sep 17 00:00:00 2001 From: Casper Date: Sun, 4 Aug 2024 18:45:06 +0200 Subject: [PATCH] ignore onnx in ignore_patterns (#570) --- awq/models/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/awq/models/base.py b/awq/models/base.py index 628c389d..4338393e 100644 --- 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: