| # Dependency set validated end-to-end on 2026-09-16. Exact pins are | |
| # intentional: this is the environment known to run both published checkpoint | |
| # families, not a broader compatibility matrix. | |
| torch==2.4.0 | |
| torchvision==0.19.0 | |
| transformers==4.57.1 | |
| diffusers==0.36.0 | |
| accelerate==1.13.0 | |
| transformer-engine[pytorch]==1.11.0 | |
| safetensors==0.7.0 | |
| tokenizers==0.22.2 | |
| huggingface-hub==0.34.0 | |
| peft==0.17.0 | |
| numpy==1.23.1 | |
| Pillow==10.4.0 | |
| requests==2.32.3 | |
| tqdm==4.67.1 | |
| typing-extensions==4.15.0 | |
| # Optional FlashAttention 2 backend (validated: flash-attn==2.7.3). The CLI | |
| # default is eager attention; --attn-implementation flash_attention_2 needs | |
| # this package. | |
| # flash-attn==2.7.3 | |