Update requirements.txt

requirements.txt

@@ -1,16 +1,30 @@
---extra-index-url https://download.pytorch.org/whl/cu117
-torch
-torchvision==0.14.1+cu117
+--extra-index-url https://download.pytorch.org/whl/cu118
+addict==2.4.0
+albumentations==1.3.0
+einops==0.6.0
+gradio==3.21
+huggingface-hub==0.13.2
+imageio==2.26.0
+imageio-ffmpeg==0.4.8
+kornia
+omegaconf==2.3.0
+open-clip-torch
+opencv-contrib-python==4.7.0.72
+opencv-python-headless==4.7.0.72
+prettytable==3.6.0
+pytorch-lightning
+safetensors==0.3.0
+timm
+https://download.pytorch.org/whl/cu118/torch-2.0.0%2Bcu118-cp39-cp39-linux_x86_64.whl
+https://download.pytorch.org/whl/cu118/torchvision-0.15.1%2Bcu118-cp39-cp39-linux_x86_64.whl
+transformers==4.27.1
+datasets==2.9.0
+accelerate
+evaluate==0.4.0
+yapf==0.32.0
+Triton==2.0.0
 #diffusers
 git+https://github.com/huggingface/diffusers.git
-#transformers
-git+https://github.com/huggingface/transformers
 scipy
 ftfy
-psutil
-accelerate
-OmegaConf
-pytorch_lightning
-triton
-#https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
-xformers
+psutil