move modeling.py and modeling_nv.py to transformers

PaddlePaddle · Dec 25, 2024 · bce5e1c · bce5e1c
1 parent bfc29f2
commit bce5e1c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git 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
@@ -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