apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Models] Add forward_meta to moe models' forward function (#5138)

Browse Source 
* [Models] Add forward_meta to moe models' forward function

* fix missing param

* fix

* fix

* fix forward_meta

* fix test and remove chunked MoE releated in config

* fix test

* fix

* fix
This commit is contained in:
Longzhi Wang
2025-12-04 13:26:58 +08:00
committed by GitHub
parent f5bdb36e9b
commit 5cd17fd662
21 changed files with 131 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/spec_decode/mtp.py
View File

@@ -988,7 +988,7 @@ class MTPProposer(Proposer):
self._get_self_hidden_states(hidden_states)
else:
if hasattr(self.model, "empty_input_forward"):
self.model.empty_input_forward()
self.model.empty_input_forward(forward_meta=self.forward_meta)
def _propose_xpu(self, step_use_cudagraph: bool = False, is_dummy_run: bool = False):
"""
@@ -1078,7 +1078,7 @@ class MTPProposer(Proposer):
self._get_self_hidden_states(hidden_states)
else:
if hasattr(self.model, "empty_input_forward"):
self.model.empty_input_forward()
self.model.empty_input_forward(self.forward_meta)
def _get_self_hidden_states(self, hidden_states):
target_hidden_states = eagle_get_self_hidden_states(