mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
[Fix] fix ernie4_5_vl model torch format loadding (#4447)
* fix * add test * fix test * fix test * update
This commit is contained in:
@@ -170,8 +170,8 @@ class Ernie4_5_VLMoeBlock(nn.Layer):
|
||||
# TODO(hehongyu): remove this after fix model network
|
||||
setattr(
|
||||
self.gate.weight,
|
||||
"model_format",
|
||||
"",
|
||||
"weight_need_transpose",
|
||||
False,
|
||||
)
|
||||
|
||||
def forward(self, hidden_states: paddle.Tensor):
|
||||
|
||||
Reference in New Issue
Block a user