| # Install with: pip install -r requirements.txt | |
| # WARNING: Do NOT run pip install torch - NV custom build will be overwritten | |
| # Core - DO NOT reinstall torch (NV custom build already installed) | |
| # torch is pre-installed as NV build 2.10.0a0+nv25.12 | |
| # flash_attn is pre-installed as 2.7.4 | |
| transformers>=4.40.0 | |
| tokenizers>=0.19.0 | |
| datasets>=2.18.0 | |
| accelerate>=0.30.0 | |
| peft>=0.10.0 | |
| trl>=0.8.6 | |
| sentencepiece>=0.1.99 | |
| wandb>=0.16.0 | |
| tensorboard>=2.16.0 | |
| PyYAML>=6.0 | |
| numpy>=1.26.0 | |
| tqdm>=4.66.0 | |