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

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
-2
View File
@@ -65,7 +65,6 @@ class RolloutModelConfig:
data_parallel_size: int = 1,
num_nextn_predict_layers: int = 0,
eplb_config: str = {},
routing_replay_config: str = None,
):
# Required parameters
self.model = model_name_or_path
@@ -114,7 +113,6 @@ class RolloutModelConfig:
self.plas_attention_config = plas_attention_config
self.num_nextn_predict_layers = num_nextn_predict_layers
self.eplb_config = eplb_config
self.routing_replay_config = routing_replay_config
def __str__(self):
return "\n".join(f"{k}: {v}" for k, v in self.__dict__.items())