From 46dabcfc194710dd504bf62df6e6080f53b58b2b Mon Sep 17 00:00:00 2001 From: baihaochen <1291308544@qq.com> Date: Sat, 23 May 2026 16:42:02 +0800 Subject: [PATCH] chore: remove unused import --- trainer/train_pretrain_deepspeed.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/trainer/train_pretrain_deepspeed.py b/trainer/train_pretrain_deepspeed.py index 63fe299..500566d 100644 --- 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')