diff --git a/torchao/quantization/GPTQ.py b/torchao/quantization/GPTQ.py index 16574b625d..c1c4842512 100644 --- a/torchao/quantization/GPTQ.py +++ b/torchao/quantization/GPTQ.py @@ -1150,6 +1150,7 @@ def _replace_linear_8da4w( padding_allowed, precision, scales_precision, + linear_class, ) def replace_linear_8da4w(