ColossalAI/examples/language/gpt/requirements.txt
Bin Jia 608cffaed3
[example] add gpt2 HybridParallelPlugin example (#4653)
* add gpt2 HybridParallelPlugin example

* update readme and testci

* update test ci

* fix test_ci bug

* update requirements

* add requirements

* update requirements

* add requirement

* rename file
2023-09-15 17:12:46 +08:00

8 lines
71 B
Plaintext

transformers >= 4.23
colossalai
evaluate
tqdm
scipy
scikit-learn
numpy