From ea58513a8fec3e3be7b36c5e8da90de6a9ccc101 Mon Sep 17 00:00:00 2001 From: ChengZi Date: Wed, 14 Dec 2022 14:38:14 +0800 Subject: [PATCH] add relative package import --- auto_transformers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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')