ColossalAI/colossalai/zero
Hongxin Liu 50793b35f4
[gemini] accelerate inference (#3641)
* [gemini] support don't scatter after inference

* [chat] update colossalai strategy

* [chat] fix opt benchmark

* [chat] update opt benchmark

* [gemini] optimize inference

* [test] add gemini inference test

* [chat] fix unit test ci

* [chat] fix ci

* [chat] fix ci

* [chat] skip checkpoint test
2023-04-26 16:32:40 +08:00
..
gemini [gemini] accelerate inference (#3641) 2023-04-26 16:32:40 +08:00
legacy [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
low_level [booster] add low level zero plugin (#3594) 2023-04-26 14:37:25 +08:00
__init__.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
wrapper.py [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00