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
prompt_dataset.py	[gemini] accelerate inference (#3641)	2023-04-26 16:32:40 +08:00
reward_dataset.py	[Coati] first commit (#3283)	2023-03-28 20:25:36 +08:00
sft_dataset.py	fix: fix sft (#3568)	2023-04-17 16:47:44 +08:00
utils.py	[Coati] first commit (#3283)	2023-03-28 20:25:36 +08:00