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
..
__init__.py	[Coati] first commit (#3283)	2023-03-28 20:25:36 +08:00
base.py	[Coati] first commit (#3283)	2023-03-28 20:25:36 +08:00
colossalai.py	[gemini] accelerate inference (#3641)	2023-04-26 16:32:40 +08:00
ddp.py	fix save_model inin naive and ddp strategy (#3436)	2023-04-04 15:30:01 +08:00
naive.py	fix save_model inin naive and ddp strategy (#3436)	2023-04-04 15:30:01 +08:00
sampler.py	[Coati] first commit (#3283)	2023-03-28 20:25:36 +08:00