Update to new version of base image
This commit is contained in:
@@ -132,7 +132,7 @@ class UnquantizedLinearMethod(LinearMethodBase):
|
||||
layer: torch.nn.Module,
|
||||
x: torch.Tensor,
|
||||
bias: Optional[torch.Tensor] = None) -> torch.Tensor:
|
||||
if (x.shape[0] == 16384 or x.shape[0] == 15360):
|
||||
if (x.shape[0] == 8192 or x.shape[0] == 16384 or x.shape[0] == 15360):
|
||||
if bias is None:
|
||||
return x @ layer.weight.T
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user