Update requirements.txt

requirements.txt

@@ -1,27 +1,64 @@
-videogen-hub
+deepspeed; platform_system != "Windows"
+bitsandbytes
+flask
+flask_restful
+flask_cors
+faiss-cpu
+fire
+h5py
+numpy>=1.23.5
+pandas<2.0.0
+peft
 torch
-huggingface-hub
-numpy
-pillow
-transformers
-gradio
-opencv-python
-ffmpeg-python
-omegaconf
-einops
-pytorch_lightning
-scipy
-tqdm
-ffmpeg
-matplotlib
+torchvision 
+torchaudio
+xformers~=0.0.20
+jupyterlab>=4.0.2
+notebook>=6.5.4
+albumentations>=1.1.0
+opencv-python>=4.2.0
+pudb~=2019.2
+imageio>=2.14.1
+imageio-ffmpeg>=0.4.7
+pytorch-lightning>=1.5.9
+omegaconf~=2.1.1
+gradio==5.0.0
+pillow~=9.5.0
+einops>=0.4.1
+torch-fidelity>=0.3.0
+setuptools>=59.5.0
+transformers==4.37.2
+torchmetrics>=0.6.0
+lpips
+dreamsim
+image-reward
+kornia>=0.6
+diffusers>=0.18.0
+accelerate>=0.20.3
+safetensors
+datasets
+tqdm>=4.64.1
+matplotlib>=3.7.1
+taming-transformers-rom1504~=0.0.6
+madgrad>=1.1
+-e git+https://github.com/openai/CLIP.git@main#egg=clip
+dominate>=2.8.0
+-e git+https://github.com/CompVis/latent-diffusion.git#egg=latent-diffusion #ldm
+jsonargparse
+openai
+nltk~=3.8.1
+krippendorff
+statsmodels
+plotly
+fal_client
+open_clip_torch
 decord
-scikit-image
-pyyaml
-requests
-pandas
-pyarrow
-torchvision
-librosa
+huggingface_hub
+open-clip-torch-any-py3
+modelscope
+protobuf==3.20.*
+rotary_embedding_torch
+av
+natsort
 tensorboard
-pytest
-setuptools
+wandb