File size: 483 Bytes
13c5606 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | torch>=2.3.0 datasets==3.6.0 einops==0.8.1 entmax==1.3 fsspec==2025.3.0 huggingface_hub==0.36.0 hf_transfer==0.1.9 hydra-core==1.3.2 lightning==2.6.0 numpy==2.3.2 omegaconf==2.3.0 packaging==25.0 Requests==2.32.4 rich==14.2.0 scipy==1.16.3 timm==1.0.22 tokenizers==0.22.1 torchmetrics==1.8.2 tqdm==4.67.1 transformers==4.57.3 triton==3.4.0 wandb==0.23.1 nvitop==1.3.2 # flash-attn must be installed separately due to build flags: # pip install flash-attn==2.8.3 --no-build-isolation |