add relative package import

2 years ago · ea58513a8f
1 changed files with 2 additions and 2 deletions
--- 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')