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

Revert "[RL] Support Rollout Routing Replay (#5321)" (#5402)

Browse Source 
This reverts commit 96d2d4877b.
This commit is contained in:
Jiang-Jia-Jun
2025-12-05 20:19:39 +08:00
committed by GitHub
parent 94c57e4175
commit c45e064f3d
24 changed files with 24 additions and 592 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
fastdeploy/engine/args_utils.py
View File

@@ -35,7 +35,6 @@ from fastdeploy.config import (
PlasAttentionConfig,
PoolerConfig,
RouterConfig,
RoutingReplayConfig,
RunnerOption,
SpeculativeConfig,
StructuredOutputsConfig,
@@ -492,11 +491,6 @@ class EngineArgs:
Configuration for eplb.
"""
routing_replay_config: Optional[Dict[str, Any]] = None
"""
Flag to rollout routing replay(r3)
"""
def __post_init__(self):
"""
Post-initialization processing to set default tokenizer if not provided.
@@ -888,12 +882,6 @@ class EngineArgs:
default=EngineArgs.eplb_config,
help="Config of eplb.",
)
parallel_group.add_argument(
"--routing-replay-config",
type=json.loads,
default=EngineArgs.routing_replay_config,
help="Flag of rollout routing replay(r3).",
)
parallel_group.add_argument(
"--enable-chunked-moe",
action="store_true",
@@ -1247,14 +1235,6 @@ class EngineArgs:
eplb_args["enable_eplb"] = self.enable_eplb
return EPLBConfig(eplb_args)
def create_routing_repaly_config(self) -> RoutingReplayConfig:
""" """
routing_replay_args = asdict(self)
if self.routing_replay_config is not None:
for k, v in self.routing_replay_config.items():
routing_replay_args[k] = v
return RoutingReplayConfig(routing_replay_args)
def create_engine_config(self, port_availability_check=True) -> FDConfig:
"""
Create and return a Config object based on the current settings.
@@ -1298,7 +1278,6 @@ class EngineArgs:
graph_opt_cfg = self.create_graph_optimization_config()
plas_attention_config = self.create_plas_attention_config()
eplb_cfg = self.create_eplb_config()
routing_replay_config = self.create_routing_repaly_config()
router_config = RouterConfig(all_dict)
early_stop_cfg = self.create_early_stop_config()
@@ -1331,5 +1310,4 @@ class EngineArgs:
graph_opt_config=graph_opt_cfg,
plas_attention_config=plas_attention_config,
early_stop_config=early_stop_cfg,
routing_replay_config=routing_replay_config,
)