File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 5252from internlm .core .parallel .comm .zero import ParamAsyncBcastHandler
5353from internlm .core .trainer import TrainState
5454from 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
5855from internlm .model .metrics import SchedulerMetricHook
5956from internlm .model .modules .embedding import Embedding1D
6057from internlm .model .modules .linear import (
7067from internlm .model .moe import Experts , MoE
7168from internlm .model .moe .moe import Qwen2MoE
7269from 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
7373from internlm .monitor import set_env_var
7474from internlm .monitor .monitor import monitor_manager as mm
7575from internlm .solver .optimizer import (
You can’t perform that action at this time.
0 commit comments