From d008c7ab0a07c99b4a806a70ed32a1af8c0e5257 Mon Sep 17 00:00:00 2001
From: Josh XT <josh@devxt.com>
Date: Mon, 22 Jan 2024 11:27:35 -0500
Subject: [PATCH] Move install

---
 Dockerfile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index b157498..5a79e77 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,9 +9,10 @@ RUN apt-get update && apt-get upgrade -y \
 WORKDIR /app
 COPY . .
 ENV CUDA_DOCKER_ARCH=all
-RUN pip install deepspeed
+
 RUN pip install --no-cache-dir -r requirements_nvidia.txt && \
     pip install --no-cache-dir -r requirements_finetune.txt && \
-    pip install --no-cache-dir -r requirements_other.txt
+    pip install --no-cache-dir -r requirements_other.txt && \
+    pip install --no-cache-dir deepspeed
 EXPOSE 7851
 CMD ["python", "script.py"]