From 6d59eb2e70efbfccedc506f4ae012302d695d5eb Mon Sep 17 00:00:00 2001
From: fxmarty <9808326+fxmarty@users.noreply.github.com>
Date: Fri, 19 Apr 2024 15:43:28 +0000
Subject: [PATCH] revert dev only changes

---
 Dockerfile_amd | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/Dockerfile_amd b/Dockerfile_amd
index 609dddb1..6dd10326 100644
--- a/Dockerfile_amd
+++ b/Dockerfile_amd
@@ -181,7 +181,8 @@ COPY --from=builder /usr/src/target/release/text-generation-launcher /usr/local/
 
 RUN cd server && \
     make gen-server && \
-    pip install -r requirements_rocm.txt
+    pip install -r requirements_rocm.txt && \
+    pip install ".[accelerate, peft, outlines]" --no-cache-dir
 
 # AWS Sagemaker compatible image
 FROM base-copy as sagemaker
@@ -193,5 +194,5 @@ ENTRYPOINT ["./entrypoint.sh"]
 # Final image
 FROM base-copy
 
-# ENTRYPOINT ["text-generation-launcher"]
-# CMD ["--json-output"]
+ENTRYPOINT ["text-generation-launcher"]
+CMD ["--json-output"]