hailin
/
train_env_prepare
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-08-08 21:06:20 +08:00
parent 5df186d033
commit 266e9ecbb5
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dspeed/install_deepspeed_src_mamba_single_gpu.sh
View File

@ -177,6 +177,7 @@ export DS_BUILD_OPS=1
export DS_BUILD_AIO=1 export DS_BUILD_AIO=1
export DS_BUILD_FUSED_ADAM=1 export DS_BUILD_FUSED_ADAM=1
export DS_BUILD_CPU_ADAM=1 export DS_BUILD_CPU_ADAM=1
export DS_BUILD_EVOFORMER_ATTN=0
# 如需推理内核：export DS_BUILD_TRANSFORMER=1 # 如需推理内核：export DS_BUILD_TRANSFORMER=1
# 提供更详细日志便于排错 # 提供更详细日志便于排错