diff --git a/dspeed/install_deepspeed_src_mamba_single_gpu.sh b/dspeed/install_deepspeed_src_mamba_single_gpu.sh index c3bd27c..5840834 100644 --- a/dspeed/install_deepspeed_src_mamba_single_gpu.sh +++ b/dspeed/install_deepspeed_src_mamba_single_gpu.sh @@ -177,6 +177,7 @@ export DS_BUILD_OPS=1 export DS_BUILD_AIO=1 export DS_BUILD_FUSED_ADAM=1 export DS_BUILD_CPU_ADAM=1 +export DS_BUILD_EVOFORMER_ATTN=0 # 如需推理内核:export DS_BUILD_TRANSFORMER=1 # 提供更详细日志便于排错