【FIX】Change the name of sparse attn from moba to plas (#4006)
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

* 更新文档

* 【docs】 update readme (#4000)

* 更新文档

* update readme

* update docs

* 【FIX】Change the name of sparse attn from moba to plas (#3845)

* 更新文档

* 更新文档

* 更新文档

* 更新文档

* 修改moba为plas

* code style

* update ci

* code style

* update ci

* code style

---------

Co-authored-by: Jiang-Jia-Jun <163579578+Jiang-Jia-Jun@users.noreply.github.com>
This commit is contained in:
yangjianfengo1
2025-09-10 10:04:29 +08:00
committed by GitHub
parent 35b8362804
commit dfc94371ee
14 changed files with 151 additions and 151 deletions

View File

@@ -59,7 +59,7 @@ class RolloutModelConfig:
graph_optimization_config: str = None,
early_stop_config: str = None,
local_rank: int = 0,
moba_attention_config: str = None,
plas_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.moba_attention_config = moba_attention_config
self.plas_attention_config = plas_attention_config
def __str__(self):
return "\n".join(f"{k}: {v}" for k, v in self.__dict__.items())