typo in git repo for pip
Browse files- docker/Dockerfile-base +2 -2
docker/Dockerfile-base
CHANGED
@@ -64,6 +64,6 @@ COPY --from=flash-attn-builder /workspace/flash-attention/dist/flash_attn-*.whl
|
|
64 |
RUN pip3 install wheels/deepspeed-*.whl wheels/flash_attn-*.whl
|
65 |
RUN git lfs install --skip-repo
|
66 |
RUN pip3 install "peft @ git+https://github.com/huggingface/peft.git@main" \
|
67 |
-
"accelerate @ git+
|
68 |
-
"transformers @ git+
|
69 |
pip3 install awscli
|
|
|
64 |
RUN pip3 install wheels/deepspeed-*.whl wheels/flash_attn-*.whl
|
65 |
RUN git lfs install --skip-repo
|
66 |
RUN pip3 install "peft @ git+https://github.com/huggingface/peft.git@main" \
|
67 |
+
"accelerate @ git+https://github.com/huggingface/accelerate.git@main" \
|
68 |
+
"transformers @ git+https://github.com/huggingface/transformers.git@main" && \
|
69 |
pip3 install awscli
|