fix: wrong shape in KLinearMarlin.

01755a60 · Yi Pan · 4e43e8a4 · 01755a60
Unverified Commit 01755a60 authored Mar 03, 2025 by Yi Pan
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

ktransformers/operators/linear.py ktransformers/operators/linear.py +0 -1

No files found.
--- a/ktransformers/operators/linear.py
+++ b/ktransformers/operators/linear.py
@@ -347,7 +347,6 @@ class KLinearMarlin(KLinearBase):
            orig_shape[-1] = self.out_features
        if self.has_bias:
            x = x + self.bias
-        orig_shape[-1] = self.n
        return x.reshape(orig_shape).to(orig_dtype)

    def unload(self):