From 4a0d881e158d2cf73128ae08942c0d5f0abdcf99 Mon Sep 17 00:00:00 2001
From: bukejiyu <52310069+bukejiyu@users.noreply.github.com>
Date: Thu, 13 Nov 2025 15:58:01 +0800
Subject: [PATCH] update (#4985)

---
 fastdeploy/model_executor/layers/mtp_linear.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/fastdeploy/model_executor/layers/mtp_linear.py b/fastdeploy/model_executor/layers/mtp_linear.py
index 4250b611f..c51523ff1 100644
--- a/fastdeploy/model_executor/layers/mtp_linear.py
+++ b/fastdeploy/model_executor/layers/mtp_linear.py
@@ -76,7 +76,7 @@ class ParallelEHProjection(nn.Layer):
                 self.linear.weight,
                 {
                     "weight_loader": default_weight_loader(self.fd_config),
-                    "model_format": self.fd_config.model_config.model_format,
+                    "weight_need_transpose": self.fd_config.model_config.model_format == "torch",
                 },
             )
             if self.bias_key is not None:
@@ -100,7 +100,7 @@ class ParallelEHProjection(nn.Layer):
                 self.linear.weight,
                 {
                     "weight_loader": default_weight_loader(self.fd_config),
-                    "model_format": self.fd_config.model_config.model_format,
+                    "weight_need_transpose": self.fd_config.model_config.model_format == "torch",
                 },
             )
             if self.nranks > 1: