fix eb5 mtp(mix) (#6800)

This commit is contained in:
cmcamdy
2026-03-13 17:36:57 +08:00
committed by GitHub
parent 8c1a2827d3
commit 7591e0d6bc
5 changed files with 55 additions and 5 deletions
@@ -264,6 +264,10 @@ class XPUForwardMeta(ForwardMeta):
# for pd_disaggregation
kv_signal_sender: Optional[paddle.Tensor] = None
hidden_states: Optional[paddle.Tensor] = None
is_draft: bool = False
def copy_from(self, other: "XPUForwardMeta", skip_keys: Optional[list] = None):
"""
Synchronize attributes from another XPUForwardMeta object