winglian commited on
Commit
66fa751
1 Parent(s): 21b7439

add huggingface packages and awscli

Browse files
Files changed (1) hide show
  1. docker/Dockerfile-base +4 -0
docker/Dockerfile-base CHANGED
@@ -63,3 +63,7 @@ COPY --from=flash-attn-builder /workspace/flash-attention/dist/flash_attn-*.whl
63
 
64
  RUN pip3 install wheels/deepspeed-*.whl wheels/flash_attn-*.whl
65
  RUN git lfs install --skip-repo
 
 
 
 
 
63
 
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+git+https://github.com/huggingface/accelerate.git@main" \
68
+ "transformers @ git+git+https://github.com/huggingface/transformers.git@main" && \
69
+ pip3 install awscli