From 376bb1bf6f719bc796a9b1e707a19bcec54e3561 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Tue, 4 Feb 2025 23:00:57 +0000
Subject: [PATCH] Lock torch version due to breaking changes

---
 requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index f45766b8..13b4ffef 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
-torch
-torchvision
+torch==2.5.1
+torchvision==0.20.1
 safetensors
 git+https://github.com/huggingface/diffusers.git
 transformers