From 5ece44b4a8ab9fdc5edb07118e45b1b30120c8fc Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Fri, 21 Mar 2025 07:19:12 -0400 Subject: [PATCH] try with reversion of packaging/setuptools/wheel install --- cicd/Dockerfile.jinja | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/cicd/Dockerfile.jinja b/cicd/Dockerfile.jinja index da3e5a8ad..a90016ee4 100644 --- a/cicd/Dockerfile.jinja +++ b/cicd/Dockerfile.jinja @@ -31,13 +31,10 @@ RUN if [ "$NIGHTLY_BUILD" = "true" ] ; then \ sed -i 's#^datasets.*#datasets @ git+https://github.com/huggingface/datasets.git@main#' requirements.txt; \ fi -RUN pip3 install -U packaging setuptools wheel ninja RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \ - pip install --no-build-isolation deepspeed; \ - pip install --no-build-isolation -e .[flash-attn,optimizers,ray,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \ + pip install --no-build-isolation -e .[deepspeed,flash-attn,optimizers,ray,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \ else \ - pip install deepspeed; \ - pip install --no-build-isolation -e .[flash-attn,optimizers,ray] $AXOLOTL_ARGS --no-binary=deepspeed; \ + pip install --no-build-isolation -e .[deepspeed,flash-attn,optimizers,ray] $AXOLOTL_ARGS; \ fi RUN python scripts/unsloth_install.py | sh