ColossalAI/examples/language/grok-1/requirements.txt
Yuanheng Zhao 5fcd7795cd
[example] update Grok-1 inference (#5495)
* revise grok-1 example

* remove unused arg in scripts

* prevent re-installing torch

* update readme

* revert modifying colossalai requirements

* add perf

* trivial

* add tokenizer url
2024-03-24 20:24:11 +08:00

4 lines
59 B
Plaintext

torch>=2.1.0,<2.2.0
colossalai>=0.3.6
transformers==4.35.0