Skip to content

Commit 01755a6

Browse files
committed
fix: wrong shape in KLinearMarlin.
1 parent 4e43e8a commit 01755a6

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

ktransformers/operators/linear.py

Lines changed: 0 additions & 1 deletion
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)