Skip to content

Commit b9afd75

Browse files
aquagullchang-wenbin
authored andcommitted
[Fix] fix ernie4_5_vl model torch format loadding (PaddlePaddle#4447)
* fix * add test * fix test * fix test * update
1 parent 1a30fa2 commit b9afd75

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

fastdeploy/model_executor/models/ernie4_5_vl/ernie4_5_vl_moe.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,8 +170,8 @@ def __init__(
170170
# TODO(hehongyu): remove this after fix model network
171171
setattr(
172172
self.gate.weight,
173-
"model_format",
174-
"",
173+
"weight_need_transpose",
174+
False,
175175
)
176176

177177
def forward(self, hidden_states: paddle.Tensor):

0 commit comments

Comments
 (0)