gpt-pretrain/requirements.txt

6 lines
93 B
Plaintext

datasets==2.11.0
deepspeed==0.9.1
pytorch-lightning==2.0.2
torch==2.0.0
transformers==4.28.1