Spaces:
Running
on
Zero
Running
on
Zero
| # Common | |
| einops==0.7.0 # Tensor operations | |
| # Training | |
| torch==2.4.0 # Torch | |
| omegaconf==2.3.0 # 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 | |
| diffusers==0.29.1 # Diffusers | |
| 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, < 25 # Code formatting | |
| flake8 >= 7, < 8 # Code style | |
| isort >= 5, < 6 # 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 # Process audio and video |