diff --git a/docs/requirements.txt b/docs/requirements.txt index 66a76122d7f..25fcdb0f00a 100644 --- a/docs/requirements.txt +++ b/docs/requirements.txt @@ -28,4 +28,5 @@ sphinxcontrib-serializinghtml==1.1.3 urllib3==1.25.3 sphinx-markdown-tables==0.0.9 numpy==1.17.2 -tensorflow==2.0.0rc2 \ No newline at end of file +tensorflow==2.0.0rc2 +torch==1.1.0 \ No newline at end of file diff --git a/docs/source/model_doc/xlm.rst b/docs/source/model_doc/xlm.rst index 157f8d71e48..786f2fb7d93 100644 --- a/docs/source/model_doc/xlm.rst +++ b/docs/source/model_doc/xlm.rst @@ -51,19 +51,19 @@ XLM ``TFXLMWithLMHeadModel`` ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ -.. autoclass:: pytorch_transformers.TFXLMWithLMHeadModel +.. autoclass:: transformers.TFXLMWithLMHeadModel :members: ``TFXLMForSequenceClassification`` ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ -.. autoclass:: pytorch_transformers.TFXLMForSequenceClassification +.. autoclass:: transformers.TFXLMForSequenceClassification :members: ``TFXLMForQuestionAnsweringSimple`` ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ -.. autoclass:: pytorch_transformers.TFXLMForQuestionAnsweringSimple +.. autoclass:: transformers.TFXLMForQuestionAnsweringSimple :members: