Lock torch version due to breaking changes

2025-02-04 23:00:57 +00:00
parent 216ab164ce
commit 376bb1bf6f
1 changed files with 2 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
-torch
+torch==2.5.1
-torchvision
+torchvision==0.20.1
 safetensors
 git+https://github.com/huggingface/diffusers.git
 transformers