Revert "【FIX】Change the name of sparse attn from moba to plas (#3845)" (#4001)
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
Deploy GitHub Pages / deploy (push) Has been cancelled

This reverts commit e31c8f7336.
This commit is contained in:
Jiang-Jia-Jun
2025-09-09 11:08:23 +08:00
committed by GitHub
parent bbd548ceb6
commit c60adf4281
13 changed files with 150 additions and 150 deletions

View File

@@ -59,7 +59,7 @@ class RolloutModelConfig:
graph_optimization_config: str = None,
early_stop_config: str = None,
local_rank: int = 0,
plas_attention_config: str = None,
moba_attention_config: str = None,
data_parallel_size: int = 1,
):
# Required parameters
@@ -106,7 +106,7 @@ class RolloutModelConfig:
self.local_rank = local_rank
self.early_stop_config = early_stop_config
self.ips = None
self.plas_attention_config = plas_attention_config
self.moba_attention_config = moba_attention_config
def __str__(self):
return "\n".join(f"{k}: {v}" for k, v in self.__dict__.items())