Browse Source

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

for more information, see https://pre-commit.ci
colossalchat
pre-commit-ci[bot] 4 months ago committed by Hongxin Liu
parent
commit
52d346f2a5
  1. 1
      tests/test_moe/test_moe_checkpoint.py

1
tests/test_moe/test_moe_checkpoint.py

@ -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

Loading…
Cancel
Save