From 66fa751c18169cb844e2727b71f137ae5fe98f25 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sun, 7 May 2023 11:51:57 -0400
Subject: [PATCH] add huggingface packages and awscli

---
 docker/Dockerfile-base | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docker/Dockerfile-base b/docker/Dockerfile-base
index 4fd65724e..5ab0677e5 100644
--- a/docker/Dockerfile-base
+++ b/docker/Dockerfile-base
@@ -63,3 +63,7 @@ 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" && \
+    pip3 install awscli