Update requirements.txt

requirements.txt

@@ -1,11 +1,8 @@
-pytorch==2.0.1
-pytorch-cuda==11.7
-torchvision==0.15.2
+
 accelerate==0.19.0
 av==10.0.0
 decord==0.6.0
 diffusers[torch]==0.16.0
-einops==0.6.1
 ffmpeg==1.4
 imageio==2.31.1
 imageio-ffmpeg==0.4.9
@@ -17,4 +14,40 @@ xformers==0.0.20
 omegaconf==2.3.0
 natsort==8.4.0
 rotary_embedding_torch
-gradio==4.3.0
+gradio==4.3.0
+
+black==23.7.0
+chardet==5.1.0
+clip @ git+https://github.com/openai/CLIP.git
+einops>=0.6.1
+fairscale>=0.4.13
+fire>=0.5.0
+fsspec>=2023.6.0
+invisible-watermark>=0.2.0
+kornia==0.6.9
+matplotlib>=3.7.2
+natsort>=8.4.0
+ninja>=1.11.1
+numpy>=1.24.4
+open-clip-torch>=2.20.0
+opencv-python==4.6.0.66
+pillow>=9.5.0
+pudb>=2022.1.3
+pytorch-lightning==2.0.1
+pyyaml>=6.0.1
+scipy>=1.10.1
+streamlit>=0.73.1
+tensorboardx==2.6
+tokenizers==0.12.1
+torch>=2.0.1
+torchaudio>=2.0.2
+torchdata==0.6.1
+torchmetrics>=1.0.1
+torchvision>=0.15.2
+triton==2.0.0
+urllib3<1.27,>=1.25.4
+wandb>=0.15.6
+webdataset>=0.2.33
+wheel>=0.41.0
+fire
+uuid