ColossalAI/examples/language/gpt/requirements.txt at db4cbdc7fb79ad2aa576d2e71759901e18cd5e1d - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 09:52:50 +00:00

Files

HELSON a7d95b7024 [example] add zero1, zero2 example in GPT examples (#2146 )

* [example] add zero1 and zero2 for GPT

* update readme in gpt example

* polish code

* change init value

* update readme

2022-12-20 14:30:27 +08:00

4 lines

58 B

Plaintext

Raw Blame History

	`colossalai >= 0.1.12`
	`torch >= 1.8.1`
	`transformers >= 4.231`