diff --git a/paddlenlp/transformers/llm_embed/modeling.py b/paddlenlp/transformers/llm_embed/modeling.py index 99475012345e..7e6933110dbf 100644 --- a/paddlenlp/transformers/llm_embed/modeling.py +++ b/paddlenlp/transformers/llm_embed/modeling.py @@ -21,9 +21,9 @@ import paddle.nn as nn from tqdm import tqdm -from ...utils.log import logger -from .. import AutoConfig, AutoModel, PretrainedModel -from ..model_outputs import ModelOutput +from paddlenlp.transformers import AutoConfig, AutoModel, PretrainedModel +from paddlenlp.transformers.model_outputs import ModelOutput +from paddlenlp.utils.log import logger @dataclass diff --git a/paddlenlp/transformers/nv_embed/modeling.py b/paddlenlp/transformers/nv_embed/modeling.py index 38a128eb9d2d..49de1c93618c 100644 --- a/paddlenlp/transformers/nv_embed/modeling.py +++ b/paddlenlp/transformers/nv_embed/modeling.py @@ -23,9 +23,14 @@ from einops import rearrange, repeat from paddle.distributed.fleet.utils import recompute -from ...utils.log import logger -from .. import AutoTokenizer, MistralModel, PretrainedConfig, PretrainedModel -from ..model_outputs import BaseModelOutputWithPast, ModelOutput +from paddlenlp.transformers import ( + AutoTokenizer, + MistralModel, + PretrainedConfig, + PretrainedModel, +) +from paddlenlp.transformers.model_outputs import BaseModelOutputWithPast, ModelOutput +from paddlenlp.utils.log import logger @dataclass