diff --git a/finetune/Dockerfile b/finetune/Dockerfile
index 2097ab3..f32dd17 100644
--- a/finetune/Dockerfile
+++ b/finetune/Dockerfile
@@ -380,9 +380,9 @@ git checkout ${DEEPSPEED_VERSION}
 # 检查 HOSTFILE_CONTENT 并写入文件
 if [ -n "${HOSTFILE_CONTENT}" ]; then
     echo "${HOSTFILE_CONTENT}" > /tmp/hostfile
-    INSTALL_CMD="./install_modified.sh ${DEEPSPEED_INSTALL_FLAGS} --hostfile /tmp/hostfile"
+    INSTALL_CMD="./install.sh ${DEEPSPEED_INSTALL_FLAGS} --hostfile /tmp/hostfile"
 else
-    INSTALL_CMD="./install_modified.sh ${DEEPSPEED_INSTALL_FLAGS}"
+    INSTALL_CMD="./install.sh ${DEEPSPEED_INSTALL_FLAGS}"
 fi
 eval $INSTALL_CMD
 EOT