Add to requirements

This commit is contained in:
Silversith 2023-04-18 11:41:37 +02:00
parent 1848db97d9
commit b9c0ccff94

View File

@ -1,11 +1,19 @@
torch torch~=2.0.0+cu118
torchdiffeq torchdiffeq~=0.2.3
torchsde torchsde~=0.2.5
einops einops~=0.6.0
open-clip-torch open-clip-torch
transformers>=4.25.1 transformers>=4.25.1
safetensors>=0.3.0 safetensors>=0.3.0
pytorch_lightning pytorch_lightning
aiohttp aiohttp~=3.8.4
accelerate accelerate~=0.18.0
pyyaml 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