fix: quant

BlinkDL · May 3, 2023 · 2e1a70e · 2e1a70e
1 parent 223488e
commit 2e1a70e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quantize/gptq/quant.py b/quantize/gptq/quant.py
@@ -148,7 +148,7 @@ def make_quant(module, names, bits, groupsize, name=''):
         make_quant(child, names, bits, groupsize, name + '.' + name1 if name != '' else name1)
 
 def make_quant_custom(module, names, bits, groupsize, name=''):
-    if isinstance(module, QuantLinear):
+    if isinstance(module, QuantLinear_custom):
         return
     for attr in dir(module):
         tmp = getattr(module, attr)