update

2024-06-21 18:46:29 +08:00
parent 8017a3e104
commit 61dd9a295d
2 changed files with 3 additions and 2 deletions
--- a/finetune/Dockerfile
+++ b/finetune/Dockerfile
@@ -371,6 +371,7 @@ git checkout .
 python setup.py bdist_wheel
 DS_BUILD_OPS=${DS_BUILD_OPS} pip install dist/deepspeed*.whl --force-reinstall
 # DS_BUILD_OPS=${DS_BUILD_OPS} pip install -r requirements/requirements.txt
+./install.sh ${DEEPSPEED_INSTALL_FLAGS} 
 # ./install.sh ${DEEPSPEED_INSTALL_FLAGS} --hostfile /job/hostfile # ./install.sh --allow_sudo --pip_sudo --no_clean --hostfile /path/to/your/hostfile
 EOT