gpt-pretrain/requirements.txt

6 lines
106 B
Plaintext
Raw Normal View History

2023-05-04 21:52:25 +08:00
datasets==2.11.0
2023-05-06 21:05:53 +08:00
pytorch-lightning[strategies]==2.0.2
tensorboardX==2.6
2023-05-04 21:52:25 +08:00
torch==2.0.0
transformers==4.28.1