diff --git a/trainer/train_pretrain_vlm.py b/trainer/train_pretrain_vlm.py
index 7b361cc..324cddc 100644
--- a/trainer/train_pretrain_vlm.py
+++ b/trainer/train_pretrain_vlm.py
@@ -16,7 +16,7 @@ from torch.utils.data import DataLoader, DistributedSampler
 from transformers import AutoTokenizer
 from model.model_vlm import MiniMindVLM, VLMConfig
 from dataset.lm_dataset import VLMDataset
-from trainer.trainer_utils import *
+from trainer.trainer_utils import get_lr, Logger, is_main_process, init_distributed_mode, setup_seed, init_vlm_model, vlm_checkpoint, SkipBatchSampler
 
 warnings.filterwarnings('ignore')
 
diff --git a/trainer/train_sft_vlm.py b/trainer/train_sft_vlm.py
index f9f6ac4..183a43f 100644
--- a/trainer/train_sft_vlm.py
+++ b/trainer/train_sft_vlm.py
@@ -16,7 +16,7 @@ from torch.utils.data import DataLoader, DistributedSampler
 from transformers import AutoTokenizer
 from model.model_vlm import MiniMindVLM, VLMConfig
 from dataset.lm_dataset import VLMDataset
-from trainer.trainer_utils import *
+from trainer.trainer_utils import get_lr, Logger, is_main_process, init_distributed_mode, setup_seed, init_vlm_model, vlm_checkpoint, SkipBatchSampler
 
 warnings.filterwarnings('ignore')