7 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	5f2e082b21	reformat code	2023-08-09 16:03:47 +08:00
zhanglei	cdf3ed9533	add residual and other moe features	2023-08-09 14:14:18 +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	8b198b2665	modified: .pre-commit-config.yaml ... modified: internlm/model/moe.py modified: internlm/model/modeling_internlm.py	2023-08-08 15:44:37 +08:00
Wenwen Qu	c357288a8b	feat(XXX): add moe	2023-08-07 20:17:49 +08:00