adding in the doc

e0849a66 · thomwolf · Lysandre Debut · 6b081f04 · e0849a66
Commit e0849a66 authored Jan 24, 2020 by thomwolf Committed by Lysandre Debut Jan 27, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

docs/source/model_doc/auto.rst docs/source/model_doc/auto.rst +7 -0

No files found.
--- a/docs/source/model_doc/auto.rst
+++ b/docs/source/model_doc/auto.rst
@@ -29,6 +29,13 @@ Instantiating one of ``AutoModel``, ``AutoConfig`` and ``AutoTokenizer`` will di
    :members:


+``AutoModelForPreTraining``
+~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: transformers.AutoModelForPreTraining
+    :members:
+
+
 ``AutoModelWithLMHead``
 ~~~~~~~~~~~~~~~~~~~~~