From b9c0ccff948ff3d687439546a90bc641c970d1c7 Mon Sep 17 00:00:00 2001 From: Silversith Date: Tue, 18 Apr 2023 11:41:37 +0200 Subject: [PATCH] Add to requirements --- requirements.txt | 22 +++++++++++++++------- 1 file changed, 15 insertions(+), 7 deletions(-) diff --git a/requirements.txt b/requirements.txt index 0527b31df..f16720a7f 100644 --- 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 \ No newline at end of file