fix rl moe gate type (#7394)

This commit is contained in:
sunxin
2026-04-14 20:04:09 +08:00
committed by GitHub
parent f6c066fb9d
commit 5f7524eb85
+2
View File
@@ -68,6 +68,7 @@ class RolloutModelConfig:
routing_replay_config: str = None,
load_choices: str = "default_v1",
lm_head_fp32: bool = False,
moe_gate_fp32: bool = True,
):
# Required parameters
self.model = model_name_or_path
@@ -121,6 +122,7 @@ class RolloutModelConfig:
self.routing_replay_config = routing_replay_config
self.load_choices = load_choices
self.lm_head_fp32 = lm_head_fp32
self.moe_gate_fp32 = moe_gate_fp32
def __str__(self):
return "\n".join(f"{k}: {v}" for k, v in self.__dict__.items())