github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/examples/language/gpt/requirements.txt

4 lines
58 B
Plaintext
Raw Normal View History

[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 06:30:27 +00:00
colossalai >= 0.1.12
[example] simplify the GPT2 huggingface example (#1826)
2022-11-08 08:14:07 +00:00
torch >= 1.8.1
transformers >= 4.231