Update requirements.txt

requirements.txt

@@ -1,22 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/cu117
-tensorflow
-protobuf==3.20.3
-torch
-torchvision==0.14.1+cu117
-#diffusers
-git+https://github.com/huggingface/diffusers.git
-#transformers
-git+https://github.com/huggingface/transformers
-scipy
-ftfy
-psutil
-accelerate==0.16.0
-#OmegaConf
-#pytorch_lightning
-triton==2.0.0.dev20230208
-#https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
-https://github.com/ZyCromerZ/xformers_builds/releases/download/xformers-2023-02-12-Py-3.8-Cuda-11.8.0-PyTorch-1.13.1%2Bcu117-ubuntu-18.04/xformers-0.0.17+12c076d.d20230212-cp38-cp38-linux_x86_64.whl
-
 addict==2.4.0
 albumentations==1.3.0
 einops==0.6.0
@@ -33,4 +14,8 @@ prettytable==3.6.0
 pytorch-lightning==1.9.0
 safetensors==0.2.8
 timm==0.6.12
-yapf==0.32.0
+torch==1.13.1
+torchvision==0.14.1
+transformers==4.26.1
+xformers==0.0.16
+yapf==0.32.0