Properly working

huggingface · Aug 9, 2024 · fa7686c · fa7686c
1 parent 757853e
commit fa7686c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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