Skip to content

Commit 64ac2f4

Browse files
authored
remove Model suffix (#2426)
1 parent 3ec1d35 commit 64ac2f4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddlenlp/transformers/auto/modeling.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,8 @@ def _from_pretrained(cls,
209209
init_kwargs = json.load(f)
210210
# class name corresponds to this configuration
211211
init_class = init_kwargs.pop("init_class", None)
212+
init_class = init_class[:-5] if init_class.endswith(
213+
"Model") else init_class
212214
if init_class:
213215
for model_flag, name in MAPPING_NAMES.items():
214216
if model_flag in init_class:

0 commit comments

Comments
 (0)