github
/
minimind
mirror of https://github.com/jingyaogong/minimind.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore: remove unused import

pull/791/head
baihaochen 2026-05-23 16:42:02 +08:00
parent 907f31f357
commit 46dabcfc19
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
trainer/train_pretrain_deepspeed.py
View File

@ -17,8 +17,7 @@ from model.model_minimind import MiniMindConfig
from model.model_minimind_mla import MiniMindMLAConfig from model.model_minimind_mla import MiniMindMLAConfig
from dataset.lm_dataset import PretrainDataset from dataset.lm_dataset import PretrainDataset
from trainer.trainer_utils import ( from trainer.trainer_utils import (
Logger, is_main_process, setup_seed, init_model, SkipBatchSampler, get_model_suffix, Logger, is_main_process, setup_seed, init_model, SkipBatchSampler, get_model_suffix
init_distributed_mode
) )
warnings.filterwarnings('ignore') warnings.filterwarnings('ignore')