From fa7686c854da8e5a5ef8278a07846e17f89ce46c Mon Sep 17 00:00:00 2001 From: "[[ -z $EMAIL ]] && read -e -p \"Enter your email (for git configuration): \" EMAIL" Date: Fri, 9 Aug 2024 14:46:03 -0400 Subject: [PATCH] Properly working --- docker/accelerate-gpu-deepspeed/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/accelerate-gpu-deepspeed/Dockerfile b/docker/accelerate-gpu-deepspeed/Dockerfile index 73d95e820ad..5c56ade5361 100644 --- a/docker/accelerate-gpu-deepspeed/Dockerfile +++ b/docker/accelerate-gpu-deepspeed/Dockerfile @@ -24,7 +24,7 @@ SHELL ["/bin/bash", "-c"] RUN source activate accelerate && conda install -c conda-forge mpi4py RUN source activate accelerate && \ python3 -m pip install --no-cache-dir \ - git+https://github.com/huggingface/accelerate#egg=accelerate[testing,test_trackers,deepspeed]@fix-deepspeed-tests \ + git+https://github.com/huggingface/accelerate@fix-deepspeed-tests#egg=accelerate[testing,test_trackers,deepspeed] \ --extra-index-url https://download.pytorch.org/whl/cu117 RUN python3 -m pip install --no-cache-dir bitsandbytes