chore: remove unused import

2026-05-23 16:42:02 +08:00 · 2026-05-23 16:42:02 +08:00 · 46dabcfc19
parent 907f31f357
commit 46dabcfc19
1 changed files with 1 additions and 2 deletions
--- a/trainer/train_pretrain_deepspeed.py
+++ b/trainer/train_pretrain_deepspeed.py
@ -17,8 +17,7 @@ from model.model_minimind import MiniMindConfig
 from model.model_minimind_mla import MiniMindMLAConfig
 from dataset.lm_dataset import PretrainDataset
 from trainer.trainer_utils import (
-    Logger, is_main_process, setup_seed, init_model, SkipBatchSampler, get_model_suffix,
-    init_distributed_mode
+    Logger, is_main_process, setup_seed, init_model, SkipBatchSampler, get_model_suffix
 )

 warnings.filterwarnings('ignore')