Set torch version to be 2.3.1 for v0.0.3

comfyanonymous · Jul 27, 2024 · 0876102 · 0876102
1 parent f87810c
commit 0876102
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
-torch
+torch==2.3.1
 torchsde
-torchvision
-torchaudio
+torchvision==0.18.1
+torchaudio==2.3.1
 einops
 transformers>=4.28.1
 tokenizers>=0.13.3