github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
colossalchat
pre-commit-ci[bot] 2024-07-19 07:54:40 +00:00 committed by Hongxin Liu
parent 46037c2ccd
commit 52d346f2a5
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_moe/test_moe_checkpoint.py
View File

@ -13,7 +13,6 @@ from transformers.models.mixtral.modeling_mixtral import MixtralForCausalLM
import colossalai import colossalai
from colossalai.booster import Booster from colossalai.booster import Booster
from colossalai.booster.plugin.moe_hybrid_parallel_plugin import MoeHybridParallelPlugin from colossalai.booster.plugin.moe_hybrid_parallel_plugin import MoeHybridParallelPlugin
from colossalai.tensor.moe_tensor.api import is_moe_tensor
from colossalai.testing import parameterize, spawn from colossalai.testing import parameterize, spawn
from colossalai.testing.utils import spawn from colossalai.testing.utils import spawn
from tests.test_moe.moe_utils import loose_close from tests.test_moe.moe_utils import loose_close