| # Core dependencies | |
| torch>=2.0.0 | |
| transformers>=4.30.0 | |
| huggingface_hub>=0.16.0 | |
| safetensors>=0.3.0 | |
| diffusers>=0.20.0 | |
| # Inference | |
| lightning>=2.0.0 | |
| ftfy | |
| # Configuration | |
| omegaconf | |
| # Utilities | |
| numpy | |
| # Note: flash-attn is required but needs special installation | |
| # See README.md for installation instructions | |