From 6a498f0d79594e2261a5773f58adf62ce1d80d7c Mon Sep 17 00:00:00 2001
From: bsilvereagle <bsilvereagle@gmail.com>
Date: Fri, 31 Mar 2023 15:51:41 -0700
Subject: [PATCH] Remove torch GPU dependencies

---
 .devops/full.Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.devops/full.Dockerfile b/.devops/full.Dockerfile
index 2b3a20c63..80752f5d4 100644
--- a/.devops/full.Dockerfile
+++ b/.devops/full.Dockerfile
@@ -6,7 +6,8 @@ RUN apt-get update && \
     apt-get install -y build-essential python3 python3-pip
 
 RUN pip install --upgrade pip setuptools wheel \
-    && pip install numpy requests sentencepiece torch tqdm
+    && pip install numpy requests sentencepiece tqdm \
+    && pip install torch==2.0.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
 
 WORKDIR /app