diff --git a/dspeed/install_deepspeed_src_mamba_single_gpu.sh b/dspeed/install_deepspeed_src_mamba_single_gpu.sh
index 9ff64f5..f9772e7 100644
--- a/dspeed/install_deepspeed_src_mamba_single_gpu.sh
+++ b/dspeed/install_deepspeed_src_mamba_single_gpu.sh
@@ -5,7 +5,7 @@ set -euo pipefail
 # Config（可用环境变量覆盖）
 # =============================
 REPO_URL="${REPO_URL:-https://github.com/microsoft/DeepSpeed.git}"
-REPO_BRANCH="${REPO_BRANCH:-v0.17.7}"
+REPO_BRANCH="${REPO_BRANCH:-v0.17.4}"
 REPO_DIR="${REPO_DIR:-DeepSpeed}"           # 将在“当前目录/REPO_DIR”下操作
 CUDA_VERSION_PIN="${CUDA_VERSION_PIN:-11.8}" # 期望 Torch 的 CUDA tag 和 nvcc 都是 11.8
 ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-}"        # 为空则自动探测