diff --git a/server/tests/utils/test_weights.py b/server/tests/utils/test_weights.py index 556fcea1..b21749d1 100644 --- a/server/tests/utils/test_weights.py +++ b/server/tests/utils/test_weights.py @@ -25,6 +25,7 @@ def gptq_weights_loader(): quant_method="gptq", quantize="gptq", sym=True, + modules_to_not_convert=[], ) @@ -37,6 +38,7 @@ def gptq_weights_loader_awq(): quant_method="awq", quantize="awq", sym=True, + modules_to_not_convert=[], )