From faeae323cb5623a39310839f0461f3b1f7d98406 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 20 Mar 2025 20:04:39 -0400 Subject: [PATCH] install deepspeed by itself --- cicd/Dockerfile.jinja | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/cicd/Dockerfile.jinja b/cicd/Dockerfile.jinja index c68076cd0..2893b06fb 100644 --- a/cicd/Dockerfile.jinja +++ b/cicd/Dockerfile.jinja @@ -33,9 +33,11 @@ RUN if [ "$NIGHTLY_BUILD" = "true" ] ; then \ RUN pip3 install -U packaging setuptools wheel RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \ - pip install --no-build-isolation -e .[deepspeed,flash-attn,optimizers,ray,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS --no-binary=deepspeed; \ + pip install deepspeed + pip install --no-build-isolation -e .[flash-attn,optimizers,ray,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \ else \ - pip install --no-build-isolation -e .[deepspeed,flash-attn,optimizers,ray] $AXOLOTL_ARGS --no-binary=deepspeed; \ + pip install deepspeed + pip install --no-build-isolation -e .[flash-attn,optimizers,ray] $AXOLOTL_ARGS --no-binary=deepspeed; \ fi RUN python scripts/unsloth_install.py | sh