ColossalAI/colossalai/legacy/moe
2024-07-19 07:32:56 +00:00
..
layer [moe] full test for deepseek and mixtral (pp + sp to fix) 2024-07-19 07:32:56 +00:00
openmoe [moe] clean legacy code 2024-07-19 07:32:01 +00:00
load_balance.py [moe] clean legacy code 2024-07-19 07:32:01 +00:00
manager.py [moe] clean legacy code 2024-07-19 07:32:01 +00:00
utils.py [moe] clean legacy code 2024-07-19 07:32:01 +00:00