From a5fe16687b896d1f7cf6edd7a6d4f32c2eefdd94 Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Wed, 28 Aug 2019 07:22:54 +0000 Subject: [PATCH] fix typo --- pytorch_transformers/modeling_auto.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_transformers/modeling_auto.py b/pytorch_transformers/modeling_auto.py index 2d28a6017f..7e65269926 100644 --- a/pytorch_transformers/modeling_auto.py +++ b/pytorch_transformers/modeling_auto.py @@ -112,7 +112,7 @@ class AutoConfig(object): """ if 'dilbert' in pretrained_model_name_or_path: - return DilBertconfig.from_pretrained(pretrained_model_name_or_path, **kwargs) + return DilBertConfig.from_pretrained(pretrained_model_name_or_path, **kwargs) elif 'roberta' in pretrained_model_name_or_path: return RobertaConfig.from_pretrained(pretrained_model_name_or_path, **kwargs) elif 'bert' in pretrained_model_name_or_path: