Add ElectraForCausalLM -> Enable Electra encoder-decoder model (#14729)

* Add ElectraForCausalLM and cover some basic tests & need to fix a few tests * Fix bugs * make style * make fix-copies * Update doc * Change docstring to markdown format * Remove redundant update_keys_to_ignore
2021-12-27 12:37:52 +01:00
parent b058490ceb
commit 501307b58b
7 changed files with 323 additions and 34 deletions
--- a/docs/source/model_doc/electra.mdx
+++ b/docs/source/model_doc/electra.mdx
@@ -83,6 +83,11 @@ This model was contributed by [lysandre](https://huggingface.co/lysandre). The o
 [[autodoc]] ElectraForPreTraining
    - forward

+## ElectraForCausalLM
+
+[[autodoc]] ElectraForCausalLM
+    - forward
+
 ## ElectraForMaskedLM

 [[autodoc]] ElectraForMaskedLM