ColossalAI/applications/Chat/coati
Hongxin Liu 7bd0bee8ea
[chat] add opt attn kernel (#3655)
* [chat] add opt attn kernel

* [chat] disable xformer during fwd
2023-05-04 16:03:33 +08:00
..
dataset [gemini] accelerate inference (#3641) 2023-04-26 16:32:40 +08:00
experience_maker [chat] polish code note typo (#3612) 2023-04-20 17:22:15 +08:00
kernels [chat] add opt attn kernel (#3655) 2023-05-04 16:03:33 +08:00
models [chat] refactor model save/load logic (#3654) 2023-04-27 18:41:49 +08:00
ray [chat] polish code note typo (#3612) 2023-04-20 17:22:15 +08:00
replay_buffer [chat] polish code note typo (#3612) 2023-04-20 17:22:15 +08:00
trainer [chat] typo accimulation_steps -> accumulation_steps (#3662) 2023-04-28 15:42:57 +08:00
utils [chat] remove lm model class (#3653) 2023-04-27 15:37:38 +08:00
__init__.py [Coati] first commit (#3283) 2023-03-28 20:25:36 +08:00