Skip to content

Commit e8c4fd6

Browse files
committed
fix ci
1 parent 5a2e996 commit e8c4fd6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

internlm/train/pipeline.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -52,9 +52,6 @@
5252
from internlm.core.parallel.comm.zero import ParamAsyncBcastHandler
5353
from internlm.core.trainer import TrainState
5454
from internlm.data.utils import unpack_type_ids
55-
from internlm.model_implementations.builder import create_model
56-
from internlm.model_implementations.registry import register_model_initializer
57-
from internlm.model_inject.inject import inject_model
5855
from internlm.model.metrics import SchedulerMetricHook
5956
from internlm.model.modules.embedding import Embedding1D
6057
from internlm.model.modules.linear import (
@@ -70,6 +67,9 @@
7067
from internlm.model.moe import Experts, MoE
7168
from internlm.model.moe.moe import Qwen2MoE
7269
from internlm.model.ops.norm import RMSNorm
70+
from internlm.model_implementations.builder import create_model
71+
from internlm.model_implementations.registry import register_model_initializer
72+
from internlm.model_inject.inject import inject_model
7373
from internlm.monitor import set_env_var
7474
from internlm.monitor.monitor import monitor_manager as mm
7575
from internlm.solver.optimizer import (

0 commit comments

Comments
 (0)