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