Fix bug
This commit is contained in:
parent
56e5bf2854
commit
132c67be0d
|
|
@ -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)))
|
||||
|
|
|
|||
Loading…
Reference in New Issue