Skip to content

Commit 91d0a1f

Browse files
Yam0214sijunhe
andauthored
fix error included by #3998 (#4201)
Co-authored-by: Sijun He <[email protected]>
1 parent 318ed29 commit 91d0a1f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

paddlenlp/transformers/ernie_m/configuration.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414
""" ERNIE-M model configuration"""
1515
from __future__ import annotations
1616

17+
from typing import Dict
18+
1719
from ..configuration_utils import PretrainedConfig
1820

1921
__all__ = ["ERNIE_M_PRETRAINED_INIT_CONFIGURATION", "ErnieMConfig", "ERNIE_M_PRETRAINED_RESOURCE_FILES_MAP"]
@@ -142,6 +144,7 @@ class ErnieMConfig(PretrainedConfig):
142144
>>> configuration = model.config
143145
```"""
144146
model_type = "ernie_m"
147+
attribute_map: Dict[str, str] = {"num_classes": "num_labels", "dropout": "classifier_dropout"}
145148
pretrained_init_configuration = ERNIE_M_PRETRAINED_INIT_CONFIGURATION
146149

147150
def __init__(

0 commit comments

Comments
 (0)