Skip to content

Commit f46b3fd

Browse files
authored
Merge pull request #778 from Conless/fix-shape
[Fix] Shape error in KLinearMarlin
2 parents 4e43e8a + 01755a6 commit f46b3fd

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

ktransformers/operators/linear.py

-1
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,6 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
347347
orig_shape[-1] = self.out_features
348348
if self.has_bias:
349349
x = x + self.bias
350-
orig_shape[-1] = self.n
351350
return x.reshape(orig_shape).to(orig_dtype)
352351

353352
def unload(self):

0 commit comments

Comments
 (0)