Set torch version to be 2.3.1 for v0.0.3

yoland68 · yoland68 · commit 1d47ec38d80b · 2024-07-26T18:54:29.000-07:00
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
@@ -13,6 +13,7 @@ Pillow
 scipy
 tqdm
 psutil
+numpy<2.0.0
 
 #non essential dependencies:
 kornia>=0.7.1
