Set torch version to be 2.3.1 for v0.0.3

2025-08-02 15:04:50 +08:00 · 2024-07-26 18:54:29 -07:00
1 changed files with 4 additions and 3 deletions
--- 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
@@ -13,6 +13,7 @@ Pillow
 scipy
 tqdm
 psutil
+numpy<2.0.0

 #non essential dependencies:
 kornia>=0.7.1