gpt-pretrain/requirements.txt

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