Added CamemBERT support for question answering.

ThilinaRajapakse · Feb 1, 2021 · 0a83ef5 · 0a83ef5
1 parent 5840749
commit 0a83ef5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/docs/_docs/16-qa-specifics.md b/docs/_docs/16-qa-specifics.md
@@ -27,6 +27,7 @@ New model types are regularly added to the library. Question Answering tasks cur
 | ----------- | --------------------------------------- |
 | ALBERT      | albert                                  |
 | BERT        | bert                                    |
+| CamemBERT   | camembert                               |
 | DistilBERT  | distilbert                              |
 | ELECTRA     | electra                                 |
 | Longformer  | longformer                              |

diff --git a/simpletransformers/question_answering/question_answering_model.py b/simpletransformers/question_answering/question_answering_model.py
@@ -46,6 +46,9 @@
     BertConfig,
     BertForQuestionAnswering,
     BertTokenizer,
+    CamembertConfig,
+    CamembertForQuestionAnswering,
+    CamembertTokenizer,
     DistilBertConfig,
     DistilBertForQuestionAnswering,
     DistilBertTokenizer,
@@ -120,6 +123,7 @@ def __init__(self, model_type, model_name, args=None, use_cuda=True, cuda_device
             "auto": (AutoConfig, AutoTokenizer, AutoModelForQuestionAnswering),
             "bart": (BartConfig, BartForQuestionAnswering, BartTokenizer),
             "bert": (BertConfig, BertForQuestionAnswering, BertTokenizer),
+            "camembert": (CamembertConfig, CamembertForQuestionAnswering, CamembertTokenizer),
             "distilbert": (DistilBertConfig, DistilBertForQuestionAnswering, DistilBertTokenizer),
             "electra": (ElectraConfig, ElectraForQuestionAnswering, ElectraTokenizer),
             "longformer": (LongformerConfig, LongformerForQuestionAnswering, LongformerTokenizer),