diff --git a/docker/Dockerfile-base b/docker/Dockerfile-base index 5ab0677e5..21716f27b 100644 --- a/docker/Dockerfile-base +++ b/docker/Dockerfile-base @@ -64,6 +64,6 @@ COPY --from=flash-attn-builder /workspace/flash-attention/dist/flash_attn-*.whl RUN pip3 install wheels/deepspeed-*.whl wheels/flash_attn-*.whl RUN git lfs install --skip-repo RUN pip3 install "peft @ git+https://github.com/huggingface/peft.git@main" \ - "accelerate @ git+git+https://github.com/huggingface/accelerate.git@main" \ - "transformers @ git+git+https://github.com/huggingface/transformers.git@main" && \ + "accelerate @ git+https://github.com/huggingface/accelerate.git@main" \ + "transformers @ git+https://github.com/huggingface/transformers.git@main" && \ pip3 install awscli