Add to requirements

2026-01-12 07:10:52 +08:00 · 2023-04-18 11:41:37 +02:00 · 2023-04-18 11:41:37 +02:00 · b9c0ccff94
commit b9c0ccff94
parent 1848db97d9
1 changed files with 15 additions and 7 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,11 +1,19 @@
-torch
-torchdiffeq
-torchsde
-einops
+torch~=2.0.0+cu118
+torchdiffeq~=0.2.3
+torchsde~=0.2.5
+einops~=0.6.0
 open-clip-torch
 transformers>=4.25.1
 safetensors>=0.3.0
 pytorch_lightning
-aiohttp
-accelerate
-pyyaml
+aiohttp~=3.8.4
+accelerate~=0.18.0
+pyyaml~=6.0
+opencv-python~=4.7.0.72
+numpy~=1.24.2
+tqdm~=4.65.0
+torchvision~=0.15.1
+timm~=0.6.13
+scipy~=1.10.1
+Pillow~=9.5.0
+psutil~=5.9.4