| gradio | |
| opencv-python-headless | |
| pillow>=10.4.0 | |
| numpy==1.24.4 | |
| git+https://github.com/huggingface/diffusers.git@main | |
| safetensors>=0.4.5 | |
| einops>=0.8.0 | |
| pyyaml>=6.0.2 | |
| omegaconf==2.3.0 | |
| mediapy | |
| librosa | |
| moviepy | |
| OmegaConf | |
| beartype | |
| sentencepiece | |
| 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 | |
| #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 |