diff --git a/docs/source/model_doc/auto.rst b/docs/source/model_doc/auto.rst index 4c76d2ab..541d03a8 100644 --- 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`` ~~~~~~~~~~~~~~~~~~~~~