We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3ec1d35 commit 64ac2f4Copy full SHA for 64ac2f4
paddlenlp/transformers/auto/modeling.py
@@ -209,6 +209,8 @@ def _from_pretrained(cls,
209
init_kwargs = json.load(f)
210
# class name corresponds to this configuration
211
init_class = init_kwargs.pop("init_class", None)
212
+ init_class = init_class[:-5] if init_class.endswith(
213
+ "Model") else init_class
214
if init_class:
215
for model_flag, name in MAPPING_NAMES.items():
216
if model_flag in init_class:
0 commit comments