InternLM/internlm/moe
Wenwen Qu fe0c342f9d get moe setting from gpc 2024-01-09 15:26:13 +08:00
..
__init__.py refactor code 2024-01-08 16:03:55 +08:00
base_moe.py refactor code 2024-01-08 16:23:53 +08:00
experts.py refactor code 2024-01-08 16:03:55 +08:00
sharded_moe.py get moe setting from gpc 2024-01-09 15:26:13 +08:00
utils.py move all2all to utils 2024-01-08 13:16:17 +08:00