From e0849a66accda8aa435a3db164c373175115a5b0 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Fri, 24 Jan 2020 17:54:32 -0500 Subject: [PATCH] adding in the doc --- docs/source/model_doc/auto.rst | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/docs/source/model_doc/auto.rst b/docs/source/model_doc/auto.rst index 4c76d2ab96..541d03a8e5 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`` ~~~~~~~~~~~~~~~~~~~~~