picotron/requirements.txt
2024-09-25 14:19:16 +00:00

7 lines
96 B
Plaintext

torch==2.1.2
triton==2.1.0
numpy==1.26.4
datasets==2.19.1
transformers==4.41.1
debugpy-run
wandb