Unverified Commit c89bdfbe authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Reorganize repo (#8580)

* Put models in subfolders

* Styling

* Fix imports in tests

* More fixes in test imports

* Sneaky hidden imports

* Fix imports in doc files

* More sneaky imports

* Finish fixing tests

* Fix examples

* Fix path for copies

* More fixes for examples

* Fix dummy files

* More fixes for example

* More model import fixes

* Is this why you're unhappy GitHub?

* Fix imports in conver command
parent 90150733
...@@ -8,6 +8,9 @@ on: ...@@ -8,6 +8,9 @@ on:
jobs: jobs:
torch_hub_integration: torch_hub_integration:
runs-on: ubuntu-latest runs-on: ubuntu-latest
env:
# TODO quickfix but may need more investigation
ACTIONS_ALLOW_UNSECURE_COMMANDS: True
steps: steps:
# no checkout necessary here. # no checkout necessary here.
- name: Extract branch name - name: Extract branch name
......
...@@ -133,7 +133,6 @@ dmypy.json ...@@ -133,7 +133,6 @@ dmypy.json
tensorflow_code tensorflow_code
# Models # Models
models
proc_data proc_data
# examples # examples
......
...@@ -51,10 +51,10 @@ AlbertTokenizer ...@@ -51,10 +51,10 @@ AlbertTokenizer
Albert specific outputs Albert specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_albert.AlbertForPreTrainingOutput .. autoclass:: transformers.models.albert.modeling_albert.AlbertForPreTrainingOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_albert.TFAlbertForPreTrainingOutput .. autoclass:: transformers.models.albert.modeling_tf_albert.TFAlbertForPreTrainingOutput
:members: :members:
......
...@@ -44,8 +44,8 @@ Implementation Notes ...@@ -44,8 +44,8 @@ Implementation Notes
- Bart doesn't use :obj:`token_type_ids` for sequence classification. Use :class:`~transformers.BartTokenizer` or - Bart doesn't use :obj:`token_type_ids` for sequence classification. Use :class:`~transformers.BartTokenizer` or
:meth:`~transformers.BartTokenizer.encode` to get the proper splitting. :meth:`~transformers.BartTokenizer.encode` to get the proper splitting.
- The forward pass of :class:`~transformers.BartModel` will create decoder inputs (using the helper function - The forward pass of :class:`~transformers.BartModel` will create decoder inputs (using the helper function
:func:`transformers.modeling_bart._prepare_bart_decoder_inputs`) if they are not passed. This is different than some :func:`transformers.models.bart.modeling_bart._prepare_bart_decoder_inputs`) if they are not passed. This is
other modeling APIs. different than some other modeling APIs.
- Model predictions are intended to be identical to the original implementation when - Model predictions are intended to be identical to the original implementation when
:obj:`force_bos_token_to_be_generated=True`. This only works, however, if the string you pass to :obj:`force_bos_token_to_be_generated=True`. This only works, however, if the string you pass to
:func:`fairseq.encode` starts with a space. :func:`fairseq.encode` starts with a space.
...@@ -93,7 +93,7 @@ BartModel ...@@ -93,7 +93,7 @@ BartModel
.. autoclass:: transformers.BartModel .. autoclass:: transformers.BartModel
:members: forward :members: forward
.. autofunction:: transformers.modeling_bart._prepare_bart_decoder_inputs .. autofunction:: transformers.models.bart.modeling_bart._prepare_bart_decoder_inputs
BartForConditionalGeneration BartForConditionalGeneration
......
...@@ -57,10 +57,10 @@ BertTokenizerFast ...@@ -57,10 +57,10 @@ BertTokenizerFast
Bert specific outputs Bert specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_bert.BertForPreTrainingOutput .. autoclass:: transformers.models.bert.modeling_bert.BertForPreTrainingOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_bert.TFBertForPreTrainingOutput .. autoclass:: transformers.models.bert.modeling_tf_bert.TFBertForPreTrainingOutput
:members: :members:
......
...@@ -71,13 +71,13 @@ DPRReaderTokenizerFast ...@@ -71,13 +71,13 @@ DPRReaderTokenizerFast
DPR specific outputs DPR specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_dpr.DPRContextEncoderOutput .. autoclass:: transformers.models.dpr.modeling_dpr.DPRContextEncoderOutput
:members: :members:
.. autoclass:: transformers.modeling_dpr.DPRQuestionEncoderOutput .. autoclass:: transformers.models.dpr.modeling_dpr.DPRQuestionEncoderOutput
:members: :members:
.. autoclass:: transformers.modeling_dpr.DPRReaderOutput .. autoclass:: transformers.models.dpr.modeling_dpr.DPRReaderOutput
:members: :members:
......
...@@ -69,10 +69,10 @@ ElectraTokenizerFast ...@@ -69,10 +69,10 @@ ElectraTokenizerFast
Electra specific outputs Electra specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_electra.ElectraForPreTrainingOutput .. autoclass:: transformers.models.electra.modeling_electra.ElectraForPreTrainingOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_electra.TFElectraForPreTrainingOutput .. autoclass:: transformers.models.electra.modeling_tf_electra.TFElectraForPreTrainingOutput
:members: :members:
......
...@@ -65,10 +65,10 @@ FunnelTokenizerFast ...@@ -65,10 +65,10 @@ FunnelTokenizerFast
Funnel specific outputs Funnel specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_funnel.FunnelForPreTrainingOutput .. autoclass:: transformers.models.funnel.modeling_funnel.FunnelForPreTrainingOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_funnel.TFFunnelForPreTrainingOutput .. autoclass:: transformers.models.funnel.modeling_tf_funnel.TFFunnelForPreTrainingOutput
:members: :members:
......
...@@ -72,10 +72,10 @@ OpenAIGPTTokenizerFast ...@@ -72,10 +72,10 @@ OpenAIGPTTokenizerFast
OpenAI specific outputs OpenAI specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_openai.OpenAIGPTDoubleHeadsModelOutput .. autoclass:: transformers.models.openai.modeling_openai.OpenAIGPTDoubleHeadsModelOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_openai.TFOpenAIGPTDoubleHeadsModelOutput .. autoclass:: transformers.models.openai.modeling_tf_openai.TFOpenAIGPTDoubleHeadsModelOutput
:members: :members:
......
...@@ -60,10 +60,10 @@ GPT2TokenizerFast ...@@ -60,10 +60,10 @@ GPT2TokenizerFast
GPT2 specific outputs GPT2 specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_gpt2.GPT2DoubleHeadsModelOutput .. autoclass:: transformers.models.gpt2.modeling_gpt2.GPT2DoubleHeadsModelOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_gpt2.TFGPT2DoubleHeadsModelOutput .. autoclass:: transformers.models.gpt2.modeling_tf_gpt2.TFGPT2DoubleHeadsModelOutput
:members: :members:
......
...@@ -93,29 +93,27 @@ LongformerTokenizerFast ...@@ -93,29 +93,27 @@ LongformerTokenizerFast
Longformer specific outputs Longformer specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_longformer.LongformerBaseModelOutput .. autoclass:: transformers.models.longformer.modeling_longformer.LongformerBaseModelOutput
:members: :members:
.. autoclass:: transformers.modeling_longformer.LongformerBaseModelOutputWithPooling .. autoclass:: transformers.models.longformer.modeling_longformer.LongformerBaseModelOutputWithPooling
:members: :members:
.. autoclass:: transformers.modeling_longformer.LongformerMultipleChoiceModelOutput .. autoclass:: transformers.models.longformer.modeling_longformer.LongformerMultipleChoiceModelOutput
:members: :members:
.. autoclass:: transformers.modeling_longformer.LongformerQuestionAnsweringModelOutput .. autoclass:: transformers.models.longformer.modeling_longformer.LongformerQuestionAnsweringModelOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_longformer.TFLongformerBaseModelOutput .. autoclass:: transformers.models.longformer.modeling_tf_longformer.TFLongformerBaseModelOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_longformer.TFLongformerBaseModelOutputWithPooling .. autoclass:: transformers.models.longformer.modeling_tf_longformer.TFLongformerBaseModelOutputWithPooling
:members: :members:
.. autoclass:: transformers.modeling_tf_longformer.TFLongformerQuestionAnsweringModelOutput .. autoclass:: transformers.models.longformer.modeling_tf_longformer.TFLongformerQuestionAnsweringModelOutput
:members: :members:
LongformerModel
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
LongformerModel LongformerModel
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
......
...@@ -67,19 +67,19 @@ LxmertTokenizerFast ...@@ -67,19 +67,19 @@ LxmertTokenizerFast
Lxmert specific outputs Lxmert specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_lxmert.LxmertModelOutput .. autoclass:: transformers.models.lxmert.modeling_lxmert.LxmertModelOutput
:members: :members:
.. autoclass:: transformers.modeling_lxmert.LxmertForPreTrainingOutput .. autoclass:: transformers.models.lxmert.modeling_lxmert.LxmertForPreTrainingOutput
:members: :members:
.. autoclass:: transformers.modeling_lxmert.LxmertForQuestionAnsweringOutput .. autoclass:: transformers.models.lxmert.modeling_lxmert.LxmertForQuestionAnsweringOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_lxmert.TFLxmertModelOutput .. autoclass:: transformers.models.lxmert.modeling_tf_lxmert.TFLxmertModelOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_lxmert.TFLxmertForPreTrainingOutput .. autoclass:: transformers.models.lxmert.modeling_tf_lxmert.TFLxmertForPreTrainingOutput
:members: :members:
......
...@@ -58,10 +58,10 @@ MobileBertTokenizerFast ...@@ -58,10 +58,10 @@ MobileBertTokenizerFast
MobileBert specific outputs MobileBert specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_mobilebert.MobileBertForPreTrainingOutput .. autoclass:: transformers.models.mobilebert.modeling_mobilebert.MobileBertForPreTrainingOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_mobilebert.TFMobileBertForPreTrainingOutput .. autoclass:: transformers.models.mobilebert.modeling_tf_mobilebert.TFMobileBertForPreTrainingOutput
:members: :members:
......
...@@ -47,16 +47,16 @@ ProphetNetTokenizer ...@@ -47,16 +47,16 @@ ProphetNetTokenizer
ProphetNet specific outputs ProphetNet specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_prophetnet.ProphetNetSeq2SeqLMOutput .. autoclass:: transformers.models.prophetnet.modeling_prophetnet.ProphetNetSeq2SeqLMOutput
:members: :members:
.. autoclass:: transformers.modeling_prophetnet.ProphetNetSeq2SeqModelOutput .. autoclass:: transformers.models.prophetnet.modeling_prophetnet.ProphetNetSeq2SeqModelOutput
:members: :members:
.. autoclass:: transformers.modeling_prophetnet.ProphetNetDecoderModelOutput .. autoclass:: transformers.models.prophetnet.modeling_prophetnet.ProphetNetDecoderModelOutput
:members: :members:
.. autoclass:: transformers.modeling_prophetnet.ProphetNetDecoderLMOutput .. autoclass:: transformers.models.prophetnet.modeling_prophetnet.ProphetNetDecoderLMOutput
:members: :members:
ProphetNetModel ProphetNetModel
......
...@@ -50,10 +50,10 @@ RagTokenizer ...@@ -50,10 +50,10 @@ RagTokenizer
Rag specific outputs Rag specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_rag.RetrievAugLMMarginOutput .. autoclass:: transformers.models.rag.modeling_rag.RetrievAugLMMarginOutput
:members: :members:
.. autoclass:: transformers.modeling_rag.RetrievAugLMOutput .. autoclass:: transformers.models.rag.modeling_rag.RetrievAugLMOutput
:members: :members:
RagRetriever RagRetriever
......
...@@ -49,16 +49,16 @@ TransfoXLTokenizer ...@@ -49,16 +49,16 @@ TransfoXLTokenizer
TransfoXL specific outputs TransfoXL specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_transfo_xl.TransfoXLModelOutput .. autoclass:: transformers.models.transfo_xl.modeling_transfo_xl.TransfoXLModelOutput
:members: :members:
.. autoclass:: transformers.modeling_transfo_xl.TransfoXLLMHeadModelOutput .. autoclass:: transformers.models.transfo_xl.modeling_transfo_xl.TransfoXLLMHeadModelOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_transfo_xl.TFTransfoXLModelOutput .. autoclass:: transformers.models.transfo_xl.modeling_tf_transfo_xl.TFTransfoXLModelOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_transfo_xl.TFTransfoXLLMHeadModelOutput .. autoclass:: transformers.models.transfo_xl.modeling_tf_transfo_xl.TFTransfoXLLMHeadModelOutput
:members: :members:
......
...@@ -50,7 +50,7 @@ XLMTokenizer ...@@ -50,7 +50,7 @@ XLMTokenizer
XLM specific outputs XLM specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_xlm.XLMForQuestionAnsweringOutput .. autoclass:: transformers.models.xlm.modeling_xlm.XLMForQuestionAnsweringOutput
:members: :members:
......
...@@ -53,43 +53,43 @@ XLNetTokenizer ...@@ -53,43 +53,43 @@ XLNetTokenizer
XLNet specific outputs XLNet specific outputs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.modeling_xlnet.XLNetModelOutput .. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetModelOutput
:members: :members:
.. autoclass:: transformers.modeling_xlnet.XLNetLMHeadModelOutput .. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetLMHeadModelOutput
:members: :members:
.. autoclass:: transformers.modeling_xlnet.XLNetForSequenceClassificationOutput .. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForSequenceClassificationOutput
:members: :members:
.. autoclass:: transformers.modeling_xlnet.XLNetForMultipleChoiceOutput .. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForMultipleChoiceOutput
:members: :members:
.. autoclass:: transformers.modeling_xlnet.XLNetForTokenClassificationOutput .. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForTokenClassificationOutput
:members: :members:
.. autoclass:: transformers.modeling_xlnet.XLNetForQuestionAnsweringSimpleOutput .. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForQuestionAnsweringSimpleOutput
:members: :members:
.. autoclass:: transformers.modeling_xlnet.XLNetForQuestionAnsweringOutput .. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForQuestionAnsweringOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetModelOutput .. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetModelOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetLMHeadModelOutput .. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetLMHeadModelOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetForSequenceClassificationOutput .. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetForSequenceClassificationOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetForMultipleChoiceOutput .. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetForMultipleChoiceOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetForTokenClassificationOutput .. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetForTokenClassificationOutput
:members: :members:
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetForQuestionAnsweringSimpleOutput .. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetForQuestionAnsweringSimpleOutput
:members: :members:
......
...@@ -21,7 +21,7 @@ import torch.nn as nn ...@@ -21,7 +21,7 @@ import torch.nn as nn
from torch.nn import CrossEntropyLoss, MSELoss from torch.nn import CrossEntropyLoss, MSELoss
from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_model_forward from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_model_forward
from transformers.modeling_albert import ( from transformers.models.albert.modeling_albert import (
ALBERT_INPUTS_DOCSTRING, ALBERT_INPUTS_DOCSTRING,
ALBERT_START_DOCSTRING, ALBERT_START_DOCSTRING,
AlbertModel, AlbertModel,
......
...@@ -23,7 +23,7 @@ from torch import nn ...@@ -23,7 +23,7 @@ from torch import nn
from torch.nn import CrossEntropyLoss, MSELoss from torch.nn import CrossEntropyLoss, MSELoss
from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_model_forward from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_model_forward
from transformers.modeling_bert import ( from transformers.models.bert.modeling_bert import (
BERT_INPUTS_DOCSTRING, BERT_INPUTS_DOCSTRING,
BERT_START_DOCSTRING, BERT_START_DOCSTRING,
BertEncoder, BertEncoder,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment