From cf5757e68b0e39d711b3994136ecb3574e355668 Mon Sep 17 00:00:00 2001
From: Dhruv Nair <dhruv.nair@gmail.com>
Date: Thu, 5 Oct 2023 10:55:09 +0000
Subject: [PATCH] pin torch version

---
 docker/diffusers-pytorch-cuda/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile
index fab3b7082765..4c447749da7b 100644
--- a/docker/diffusers-pytorch-cuda/Dockerfile
+++ b/docker/diffusers-pytorch-cuda/Dockerfile
@@ -25,8 +25,8 @@ ENV PATH="/opt/venv/bin:$PATH"
 # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
 RUN python3 -m pip install --no-cache-dir --upgrade pip && \
     python3 -m pip install --no-cache-dir \
-        torch \
-        torchvision \
+        torch==2.0.1 \
+        torchvision==0.15.2 \
         torchaudio \
         invisible_watermark && \
     python3 -m pip install --no-cache-dir \