Update requirements.txt

requirements.txt

@@ -1,24 +1,55 @@
-bark_ssg==1.3.4
+accelerate==0.19.0
+av==10.0.0
 decord==0.6.0
-diffusers==0.25.0
-einops==0.7.0
-imageio==2.28.0
-ipython==8.14.0
-librosa==0.10.1
-mmcv==2.1.0
-moviepy==1.0.3
-natsort==8.3.1
-nltk==3.8.1
-numpy==1.23.5
-omegaconf==2.3.0
-openai==0.27.8
-opencv_python==4.7.0.72
-Pillow==9.4.0
-pytorch_lightning==2.0.2
-rotary_embedding_torch==0.2.3
-soundfile==0.12.1
-torch==2.0.0
-torchvision==0.15.0
+diffusers[torch]==0.16.0
+ffmpeg==1.4
+imageio==2.31.1
+imageio-ffmpeg==0.4.9
+pandas==2.0.1
+timm==0.6.13
 tqdm==4.65.0
 transformers==4.28.1
-xformers==0.0.19
+xformers==0.0.20
+omegaconf==2.3.0
+natsort==8.4.0
+rotary_embedding_torch
+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
+python-docx
+tensorboardX
+tensorboard