diff --git a/simpletransformers/custom_models/models.py b/simpletransformers/custom_models/models.py index 52cbf85c..5e342e64 100755 --- a/simpletransformers/custom_models/models.py +++ b/simpletransformers/custom_models/models.py @@ -14,6 +14,7 @@ XLNetModel, XLNetPreTrainedModel, ) + from transformers.configuration_camembert import CamembertConfig from transformers.configuration_distilbert import DistilBertConfig from transformers.configuration_roberta import RobertaConfig