From a20969170be8ded82593b034209aef0b76e2d54f Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Date: Mon, 20 Jul 2020 17:53:21 -0400
Subject: [PATCH] Add AlbertForPretraining to doc (#5914)

---
 docs/source/model_doc/albert.rst | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/docs/source/model_doc/albert.rst b/docs/source/model_doc/albert.rst
index ab382a27c7..6562631e0a 100644
--- a/docs/source/model_doc/albert.rst
+++ b/docs/source/model_doc/albert.rst
@@ -61,6 +61,13 @@ AlbertModel
     :members:
 
 
+AlbertForPreTraining
+~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: transformers.AlbertForPreTraining
+    :members:
+
+
 AlbertForMaskedLM
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
@@ -103,6 +110,13 @@ TFAlbertModel
     :members:
 
 
+TFAlbertForPreTraining
+~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: transformers.TFAlbertForPreTraining
+    :members:
+
+
 TFAlbertForMaskedLM
 ~~~~~~~~~~~~~~~~~~~~~~~~~~