gpt-pretrain/requirements.txt

6 lines
93 B
Plaintext
Raw Normal View History

2023-05-04 21:52:25 +08:00
datasets==2.11.0
deepspeed==0.9.1
pytorch-lightning==2.0.2
torch==2.0.0
transformers==4.28.1