From 7e957237e4247ca222096f6120eea2359b3012d1 Mon Sep 17 00:00:00 2001 From: LysandreJik Date: Thu, 26 Sep 2019 10:08:56 -0400 Subject: [PATCH] [Doc] XLM + Torch in documentation --- docs/requirements.txt | 3 ++- docs/source/model_doc/xlm.rst | 6 +++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/docs/requirements.txt b/docs/requirements.txt index 66a76122d7..25fcdb0f00 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 157f8d71e4..786f2fb7d9 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: