diff --git a/autograd_4bit.py b/autograd_4bit.py index be7d73f..5d19908 100644 --- a/autograd_4bit.py +++ b/autograd_4bit.py @@ -107,7 +107,6 @@ class Autograd4bitQuantLinear(nn.Module): self.bits = bits self.maxq = 2 ** self.bits - 1 self.groupsize = groupsize - self.g_idx = 0 if groupsize == -1: self.register_buffer('zeros', torch.empty((out_features, 1))) self.register_buffer('scales', torch.empty((out_features, 1)))