[MTP]Support RL reshard (#4074)
Some checks failed
CE Compile Job / ce_job_pre_check (push) Has been cancelled
CE Compile Job / print_ce_job_pre_check_outputs (push) Has been cancelled
CE Compile Job / FD-Clone-Linux (push) Has been cancelled
CE Compile Job / Show Code Archive Output (push) Has been cancelled
CE Compile Job / BUILD_SM8090 (push) Has been cancelled
CE Compile Job / BUILD_SM8689 (push) Has been cancelled
CE Compile Job / CE_UPLOAD (push) Has been cancelled

* support rl reshard

* modify model name
This commit is contained in:
freeliuzc
2025-09-15 11:47:06 +08:00
committed by GitHub
parent 2883746132
commit d2ab369427
3 changed files with 4 additions and 0 deletions

View File

@@ -132,6 +132,7 @@ class ModelConfig:
self.eos_tokens_lens: int = 2
self.lm_head_fp32: bool = False
self.model_format = "auto"
self.num_nextn_predict_layers = 0
for key, value in args.items():
if hasattr(self, key):
setattr(self, key, value)