picotron/requirements.txt

6 lines
90 B
Plaintext

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