5 Commits (0e6b1f856cbc71b4926d975a55b0ae3e80a1d46d)

Author	SHA1	Message	Date
Wenwen Qu	0e6b1f856c	add support for moe checkpoint	2023-08-24 17:01:14 +08:00
Wenwen Qu	929dd36cf2	avoid moe parameter partition in zero optimizer	2023-08-15 17:52:51 +08:00
Wenwen Qu	9c8b999291	modified: internlm/model/moe.py ... modified: internlm/moe/sharded_moe.py modified: internlm/utils/parallel.py	2023-08-08 16:46:14 +08:00
Wenwen Qu	5b6cf7cab0	reformat code	2023-08-08 15:07:04 +08:00
Wenwen Qu	c357288a8b	feat(XXX): add moe	2023-08-07 20:17:49 +08:00