diff --git a/auto_transformers.py b/auto_transformers.py
index 074d1d7..ae949b0 100644
--- a/auto_transformers.py
+++ b/auto_transformers.py
@@ -26,8 +26,8 @@ from towhee import register
 import warnings
 import logging
 
-from train_mlm_with_hf_trainer import train_mlm_with_hf_trainer
-from train_clm_with_hf_trainer import train_clm_with_hf_trainer
+from .train_mlm_with_hf_trainer import train_mlm_with_hf_trainer
+from .train_clm_with_hf_trainer import train_clm_with_hf_trainer
 
 log = logging.getLogger('run_op')