From 112aae1e026e9f264cde8526cd130a96ac3fa0ca Mon Sep 17 00:00:00 2001
From: davidmezzetti <561939+davidmezzetti@users.noreply.github.com>
Date: Thu, 5 Sep 2024 20:54:06 -0400
Subject: [PATCH] Rebuild Docker images for txtai 7.4, closes #14

---
 Dockerfile       | 4 ++--
 requirements.txt | 5 +++++
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 2892716..96d8e29 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,8 +12,8 @@ RUN \
     rm -rf /var/lib/apt/lists && \
     apt-get -y autoremove && \
     \
-    # Install base requirements, latest Transformers (Llama 3.1)
-    python -m pip install --no-cache-dir -r requirements.txt transformers --upgrade
+    # Install base requirements
+    python -m pip install --no-cache-dir -r requirements.txt
 
 # Start streamlit application
 ENTRYPOINT ["streamlit", "run", "rag.py"]
diff --git a/requirements.txt b/requirements.txt
index a233029..b5ade2d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,9 @@
+# Current version of AutoAWQ requires torch==2.3.1
 autoawq; platform_machine == "x86_64" or platform_machine == "AMD64"
+torch==2.3.1; platform_machine == "x86_64" or platform_machine == "AMD64"
+torchvision==0.18.1; platform_machine == "x86_64" or platform_machine == "AMD64"
+
+# Base requirements
 matplotlib
 streamlit
 txtai[graph,pipeline-data,pipeline-llm]