Update requirements.txt

requirements.txt

@@ -1,15 +1,96 @@
-accelerate
-transformers
+gradio
+opencv-python-headless
+pillow>=10.4.0
+numpy==1.24.4
+
+safetensors>=0.4.5
+einops>=0.8.0
+pyyaml>=6.0.2
+omegaconf==2.3.0 
+mediapy
+librosa
+moviepy
+OmegaConf
+beartype
 sentencepiece
-pillow
-numpy
-torchvision
-huggingface_hub
-spaces
-opencv-python
-imageio
-imageio-ffmpeg
+gradio[oauth]
+mediapy
+fastapi
+uvicorn[standard]
+tiktoken
+transformers_stream_generator
+rotary-embedding-torch
+transformers
+accelerate
+safetensors
 einops
+decord
+huggingface_hub
+google-generativeai
+gradio
+tabulate
+pydantic
+soundfile
+requests
+hf_transfer
 timm
+ftfy
+easydict
 av
-git+https://github.com/huggingface/diffusers.git@main
+
+
+#vince
+bitsandbytes==0.45.0
+einop
+numpy
+rotary-embedding-torch
+safetensors
+sentencepiece
+torch
+torchvision
+transformers
+
+
+
+Common
+                # Configuration system.
+lpips>=0.1.4                    # LPIPS model for VAE training
+
+# Dataloading
+bson>=0.5.10                    # bson for Image dataloading
+tensorflow>=2.16.1              # Video dataloading
+opencv-python>=4.9.0.80         # OpenCV
+sentencepiece>=0.2.0            # For Text Encoder
+
+# Modeling
+rotary-embedding-torch>=0.5.3   # Rotary positional embedding
+transformers>=4.38.2            # Transformers
+torchvision>=0.19.0             # Torchvision
+tiktoken>=0.7.0                 # Tiktoken for generation
+transformers_stream_generator>=0.0.5 #LLM generation support
+
+# Metrics
+torchmetrics>=1.3.2             # Core module for metric
+pycocotools>=2.0.7              # COCO-related
+torch-fidelity>=0.3.0           # FID-related
+
+# Experiment Tracking
+moviepy>=1.0.3                  # WandB Logging Image & Video
+imageio>=2.34.0                 # WandB Logging Image & Video
+tabulate>=0.9.0                 # Logging Table
+deepdiff>=7.0.1                 # Find difference of config
+
+# Testing
+parameterized>=0.9.0            # Define multiple tests through decorators.
+
+# Notebook
+mediapy>=1.2.0                  # Notebook Visualization
+
+# DevOPs
+black >= 24             # Code formatting
+flake8 >= 7              # Code style
+isort >= 5               # Import sorting
+pre-commit>=3.7.0               # Pre-commit hooks
+expecttest>=0.2.1               # Pytorch dist tests
+hypothesis>=6.100.1             # Fix randomness
+av>=12.0.0