Skip to content

Commit c6ede30

Browse files
committed
fix bug
Signed-off-by: funanyang <[email protected]> Co-authored-by: funanyang [email protected]
1 parent 06241d9 commit c6ede30

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/torchair/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ def torchair_ops_patch():
208208
from vllm_ascend.torchair.ops.torchair_linear import (
209209
torchair_oproj_tp_forward)
210210

211-
AscendRowParallelLinear.forward = torchair_oproj_tp_forward # type: ignore[method-assign]
211+
AscendRowParallelLinear._forward_oproj_tp = torchair_oproj_tp_forward # type: ignore[method-assign]
212212

213213
AscendRotaryEmbedding.__init__ = qwen_rope_init_func # type: ignore[method-assign]
214214
AscendRotaryEmbedding.forward_oot = rope_forward # type: ignore[method-assign]

0 commit comments

Comments
 (0)