From 01755a60c055d194c2851a57edc0536bbee44f08 Mon Sep 17 00:00:00 2001
From: Yi Pan <conlesspan@outlook.com>
Date: Mon, 3 Mar 2025 17:34:37 +0800
Subject: [PATCH] fix: wrong shape in KLinearMarlin.

---
 ktransformers/operators/linear.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/ktransformers/operators/linear.py b/ktransformers/operators/linear.py
index 96d3578..103fc1a 100644
--- 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):