apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[MTP] support expert-parellel in mtp (#2835)

Browse Source
This commit is contained in:
freeliuzc
2025-07-14 14:28:50 +08:00
committed by GitHub
parent ece88596ed
commit 7f64d408a9
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
fastdeploy/model_executor/load_weight_utils.py
View File

@@ -250,7 +250,8 @@ def load_composite_checkpoint(
# 2. Tensor Parallel (TP)
# 3. Pre-sharded (pre-split)
"""
if fd_config.parallel_config.use_ep:
if fd_config.parallel_config.use_ep and \
fd_config.speculative_config.model_type != "mtp":
state_dict = load_ep_checkpoint(model_path,
fd_config.model_config,
return_numpy=True)