[Doc] XLM + Torch in documentation

7e957237 · LysandreJik · 302a4813 · 7e957237 · 7e957237
Commit 7e957237 authored Sep 26, 2019 by LysandreJik
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 4 deletions

docs/requirements.txt docs/requirements.txt +2 -1

docs/source/model_doc/xlm.rst docs/source/model_doc/xlm.rst +3 -3

No files found.
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -29,3 +29,4 @@ urllib3==1.25.3
 sphinx-markdown-tables==0.0.9
 numpy==1.17.2
 tensorflow==2.0.0rc2
+torch==1.1.0
\ No newline at end of file
--- 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: