| # Core deep learning |
| torch==2.7.1 |
| torchvision==0.22.1 |
| torchdiffeq==0.2.5 |
| torchsde==0.2.6 |
|
|
| # Diffusion / generation |
| diffusers==0.33.1 |
| accelerate==1.12.0 |
| transformers==4.57.1 |
| safetensors==0.6.2 |
| peft==0.17.1 |
|
|
| # Training utilities |
| deepspeed==0.17.6 |
| came-pytorch==0.1.3 |
| tensorboard==2.20.0 |
|
|
| # Vision / video |
| opencv-python==4.10.0.84 |
| scikit-image==0.25.2 |
| imageio==2.37.0 |
| imageio-ffmpeg==0.6.0 |
| mediapy==1.2.4 |
| decord==0.6.0 |
| kornia==0.8.1 |
| albumentations==2.0.8 |
| timm==1.0.19 |
| tomesd==0.1.3 |
| Pillow==11.3.0 |
|
|
| # Data / ML utilities |
| numpy==1.26.4 |
| scipy==1.14.0 |
| scikit-learn==1.7.2 |
| datasets==4.0.0 |
| einops==0.8.0 |
|
|
| # Config / logging |
| omegaconf==2.3.0 |
| ml_collections==1.1.0 |
| absl-py==2.3.1 |
| loguru==0.7.3 |
| tqdm==4.67.1 |
| matplotlib==3.10.6 |
|
|
| # NLP |
| sentencepiece==0.2.1 |
| ftfy==6.1.1 |
| beautifulsoup4==4.13.5 |
|
|
| # Misc |
| func-timeout==4.3.5 |
| requests==2.32.5 |
| packaging==25.0 |
|
|
| # Optional: Gradio UI (only needed for app.py / demo) |
| # gradio>=3.41.2,<=3.48.0 |
|
|
| # Note: SAM2 must be installed separately per the instructions at |
| # https://github.com/facebookresearch/sam2?tab=readme-ov-file#installation |
|
|
