Skip to content

Commit 920ce29

Browse files
[Fluid] Change paddle.fluid.layers.utils to paddle.utils (#5121)
1 parent 9989d60 commit 920ce29

File tree

4 files changed

+20
-4
lines changed

4 files changed

+20
-4
lines changed

paddlenlp/transformers/generation_utils.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,11 @@
2121
import paddle.nn as nn
2222
import paddle.nn.functional as F
2323
from paddle.common_ops_import import convert_dtype
24-
from paddle.fluid.layers.utils import map_structure
24+
25+
try:
26+
from paddle.utils import map_structure
27+
except ImportError:
28+
from paddle.fluid.layers.utils import map_structure
2529

2630
from paddlenlp.utils.log import logger
2731

paddlenlp/transformers/transformer/modeling.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,11 @@
1616
import paddle
1717
import paddle.nn as nn
1818
import paddle.nn.functional as F
19-
from paddle.fluid.layers.utils import map_structure
19+
20+
try:
21+
from paddle.utils import map_structure
22+
except ImportError:
23+
from paddle.fluid.layers.utils import map_structure
2024
from paddle.nn import (
2125
TransformerDecoder,
2226
TransformerDecoderLayer,

tests/test_tipc/transformer/modeling.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,11 @@
1616
import paddle
1717
import paddle.nn as nn
1818
import paddle.nn.functional as F
19-
from paddle.fluid.layers.utils import map_structure
19+
20+
try:
21+
from paddle.utils import map_structure
22+
except ImportError:
23+
from paddle.fluid.layers.utils import map_structure
2024

2125
__all__ = [
2226
"position_encoding_init",

tests/transformer/modeling.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,11 @@
1616
import paddle
1717
import paddle.nn as nn
1818
import paddle.nn.functional as F
19-
from paddle.fluid.layers.utils import map_structure
19+
20+
try:
21+
from paddle.utils import map_structure
22+
except ImportError:
23+
from paddle.fluid.layers.utils import map_structure
2024

2125
__all__ = [
2226
"position_encoding_init",

0 commit comments

Comments
 (0)