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]