Skip to content

GitLab

Commit a5fe1668 authored by VictorSanh's avatar VictorSanh
Browse files

fix typo

parent 497f73c9
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_transformers/modeling_auto.py
View file @ a5fe1668
...@@ -112,7 +112,7 @@ class AutoConfig(object): ...@@ -112,7 +112,7 @@ class AutoConfig(object):
""" """
if 'dilbert' in pretrained_model_name_or_path: 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: elif 'roberta' in pretrained_model_name_or_path:
return RobertaConfig.from_pretrained(pretrained_model_name_or_path, **kwargs) return RobertaConfig.from_pretrained(pretrained_model_name_or_path, **kwargs)
elif 'bert' in pretrained_model_name_or_path: elif 'bert' in pretrained_model_name_or_path:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment