support conv1d

2023-04-28 09:14:44 +09:00 · 2023-04-28 09:14:44 +09:00 · c1b7c7647d
commit c1b7c7647d
parent ac41f68532
1 changed files with 1 additions and 1 deletions
--- a/auto_gptq/quantization/gptq.py
+++ b/auto_gptq/quantization/gptq.py
@ -23,7 +23,7 @@ class GPTQ:
        W = layer.weight.data.clone()
        if isinstance(self.layer, nn.Conv2d):
            W = W.flatten(1)
-        if isinstance(self.layer, transformers.Conv1D):
+        if isinstance(self.layer, transformers.pytorch_utils.Conv1D):
            W = W.t()
        self.rows = W.shape[0]
        self.columns = W.shape[1]