Merge pull request #778 from Conless/fix-shape

[Fix] Shape error in KLinearMarlin

Merge pull request #778 from Conless/fix-shape
[Fix] Shape error in KLinearMarlin
f46b3fd5 · Atream · GitHub · 4e43e8a4 · 01755a60 · f46b3fd5
Unverified Commit f46b3fd5 authored Mar 03, 2025 by Atream Committed by GitHub Mar 03, 2025
Hide 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):