adding in the doc

This commit is contained in:
thomwolf
2020-01-24 17:54:32 -05:00
committed by Lysandre Debut
parent 6b081f04e6
commit e0849a66ac

View File

@@ -29,6 +29,13 @@ Instantiating one of ``AutoModel``, ``AutoConfig`` and ``AutoTokenizer`` will di
:members: :members:
``AutoModelForPreTraining``
~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.AutoModelForPreTraining
:members:
``AutoModelWithLMHead`` ``AutoModelWithLMHead``
~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~