Skip to content

Commit c89bdfb

Browse files
sguggersgugger
authored
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
1 parent 9015073 commit c89bdfb

File tree

381 files changed

+2555
-1475
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

381 files changed

+2555
-1475
lines changed

‎.github/workflows/github-torch-hub.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,9 @@ on:
88
jobs:
99
torch_hub_integration:
1010
runs-on: ubuntu-latest
11+
env:
12+
# TODO quickfix but may need more investigation
13+
ACTIONS_ALLOW_UNSECURE_COMMANDS: True
1114
steps:
1215
# no checkout necessary here.
1316
- name: Extract branch name

‎.gitignore

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,6 @@ dmypy.json
133133
tensorflow_code
134134

135135
# Models
136-
models
137136
proc_data
138137

139138
# examples

‎docs/source/model_doc/albert.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,10 @@ AlbertTokenizer
5151
Albert specific outputs
5252
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
5353

54-
.. autoclass:: transformers.modeling_albert.AlbertForPreTrainingOutput
54+
.. autoclass:: transformers.models.albert.modeling_albert.AlbertForPreTrainingOutput
5555
:members:
5656

57-
.. autoclass:: transformers.modeling_tf_albert.TFAlbertForPreTrainingOutput
57+
.. autoclass:: transformers.models.albert.modeling_tf_albert.TFAlbertForPreTrainingOutput
5858
:members:
5959

6060

‎docs/source/model_doc/bart.rst

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@ Implementation Notes
4444
- Bart doesn't use :obj:`token_type_ids` for sequence classification. Use :class:`~transformers.BartTokenizer` or
4545
:meth:`~transformers.BartTokenizer.encode` to get the proper splitting.
4646
- The forward pass of :class:`~transformers.BartModel` will create decoder inputs (using the helper function
47-
:func:`transformers.modeling_bart._prepare_bart_decoder_inputs`) if they are not passed. This is different than some
48-
other modeling APIs.
47+
:func:`transformers.models.bart.modeling_bart._prepare_bart_decoder_inputs`) if they are not passed. This is
48+
different than some other modeling APIs.
4949
- Model predictions are intended to be identical to the original implementation when
5050
:obj:`force_bos_token_to_be_generated=True`. This only works, however, if the string you pass to
5151
:func:`fairseq.encode` starts with a space.
@@ -93,7 +93,7 @@ BartModel
9393
.. autoclass:: transformers.BartModel
9494
:members: forward
9595

96-
.. autofunction:: transformers.modeling_bart._prepare_bart_decoder_inputs
96+
.. autofunction:: transformers.models.bart.modeling_bart._prepare_bart_decoder_inputs
9797

9898

9999
BartForConditionalGeneration

‎docs/source/model_doc/bert.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,10 +57,10 @@ BertTokenizerFast
5757
Bert specific outputs
5858
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
5959

60-
.. autoclass:: transformers.modeling_bert.BertForPreTrainingOutput
60+
.. autoclass:: transformers.models.bert.modeling_bert.BertForPreTrainingOutput
6161
:members:
6262

63-
.. autoclass:: transformers.modeling_tf_bert.TFBertForPreTrainingOutput
63+
.. autoclass:: transformers.models.bert.modeling_tf_bert.TFBertForPreTrainingOutput
6464
:members:
6565

6666

‎docs/source/model_doc/dpr.rst

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,13 +71,13 @@ DPRReaderTokenizerFast
7171
DPR specific outputs
7272
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
7373

74-
.. autoclass:: transformers.modeling_dpr.DPRContextEncoderOutput
74+
.. autoclass:: transformers.models.dpr.modeling_dpr.DPRContextEncoderOutput
7575
:members:
7676

77-
.. autoclass:: transformers.modeling_dpr.DPRQuestionEncoderOutput
77+
.. autoclass:: transformers.models.dpr.modeling_dpr.DPRQuestionEncoderOutput
7878
:members:
7979

80-
.. autoclass:: transformers.modeling_dpr.DPRReaderOutput
80+
.. autoclass:: transformers.models.dpr.modeling_dpr.DPRReaderOutput
8181
:members:
8282

8383

‎docs/source/model_doc/electra.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,10 +69,10 @@ ElectraTokenizerFast
6969
Electra specific outputs
7070
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
7171

72-
.. autoclass:: transformers.modeling_electra.ElectraForPreTrainingOutput
72+
.. autoclass:: transformers.models.electra.modeling_electra.ElectraForPreTrainingOutput
7373
:members:
7474

75-
.. autoclass:: transformers.modeling_tf_electra.TFElectraForPreTrainingOutput
75+
.. autoclass:: transformers.models.electra.modeling_tf_electra.TFElectraForPreTrainingOutput
7676
:members:
7777

7878

‎docs/source/model_doc/funnel.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,10 +65,10 @@ FunnelTokenizerFast
6565
Funnel specific outputs
6666
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
6767

68-
.. autoclass:: transformers.modeling_funnel.FunnelForPreTrainingOutput
68+
.. autoclass:: transformers.models.funnel.modeling_funnel.FunnelForPreTrainingOutput
6969
:members:
7070

71-
.. autoclass:: transformers.modeling_tf_funnel.TFFunnelForPreTrainingOutput
71+
.. autoclass:: transformers.models.funnel.modeling_tf_funnel.TFFunnelForPreTrainingOutput
7272
:members:
7373

7474

‎docs/source/model_doc/gpt.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,10 +72,10 @@ OpenAIGPTTokenizerFast
7272
OpenAI specific outputs
7373
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
7474

75-
.. autoclass:: transformers.modeling_openai.OpenAIGPTDoubleHeadsModelOutput
75+
.. autoclass:: transformers.models.openai.modeling_openai.OpenAIGPTDoubleHeadsModelOutput
7676
:members:
7777

78-
.. autoclass:: transformers.modeling_tf_openai.TFOpenAIGPTDoubleHeadsModelOutput
78+
.. autoclass:: transformers.models.openai.modeling_tf_openai.TFOpenAIGPTDoubleHeadsModelOutput
7979
:members:
8080

8181

‎docs/source/model_doc/gpt2.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,10 @@ GPT2TokenizerFast
6060
GPT2 specific outputs
6161
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
6262

63-
.. autoclass:: transformers.modeling_gpt2.GPT2DoubleHeadsModelOutput
63+
.. autoclass:: transformers.models.gpt2.modeling_gpt2.GPT2DoubleHeadsModelOutput
6464
:members:
6565

66-
.. autoclass:: transformers.modeling_tf_gpt2.TFGPT2DoubleHeadsModelOutput
66+
.. autoclass:: transformers.models.gpt2.modeling_tf_gpt2.TFGPT2DoubleHeadsModelOutput
6767
:members:
6868

6969

‎docs/source/model_doc/longformer.rst

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -93,29 +93,27 @@ LongformerTokenizerFast
9393
Longformer specific outputs
9494
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
9595

96-
.. autoclass:: transformers.modeling_longformer.LongformerBaseModelOutput
96+
.. autoclass:: transformers.models.longformer.modeling_longformer.LongformerBaseModelOutput
9797
:members:
9898

99-
.. autoclass:: transformers.modeling_longformer.LongformerBaseModelOutputWithPooling
99+
.. autoclass:: transformers.models.longformer.modeling_longformer.LongformerBaseModelOutputWithPooling
100100
:members:
101101

102-
.. autoclass:: transformers.modeling_longformer.LongformerMultipleChoiceModelOutput
102+
.. autoclass:: transformers.models.longformer.modeling_longformer.LongformerMultipleChoiceModelOutput
103103
:members:
104104

105-
.. autoclass:: transformers.modeling_longformer.LongformerQuestionAnsweringModelOutput
105+
.. autoclass:: transformers.models.longformer.modeling_longformer.LongformerQuestionAnsweringModelOutput
106106
:members:
107107

108-
.. autoclass:: transformers.modeling_tf_longformer.TFLongformerBaseModelOutput
108+
.. autoclass:: transformers.models.longformer.modeling_tf_longformer.TFLongformerBaseModelOutput
109109
:members:
110110

111-
.. autoclass:: transformers.modeling_tf_longformer.TFLongformerBaseModelOutputWithPooling
111+
.. autoclass:: transformers.models.longformer.modeling_tf_longformer.TFLongformerBaseModelOutputWithPooling
112112
:members:
113113

114-
.. autoclass:: transformers.modeling_tf_longformer.TFLongformerQuestionAnsweringModelOutput
114+
.. autoclass:: transformers.models.longformer.modeling_tf_longformer.TFLongformerQuestionAnsweringModelOutput
115115
:members:
116116

117-
LongformerModel
118-
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
119117

120118
LongformerModel
121119
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

‎docs/source/model_doc/lxmert.rst

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -67,19 +67,19 @@ LxmertTokenizerFast
6767
Lxmert specific outputs
6868
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
6969

70-
.. autoclass:: transformers.modeling_lxmert.LxmertModelOutput
70+
.. autoclass:: transformers.models.lxmert.modeling_lxmert.LxmertModelOutput
7171
:members:
7272

73-
.. autoclass:: transformers.modeling_lxmert.LxmertForPreTrainingOutput
73+
.. autoclass:: transformers.models.lxmert.modeling_lxmert.LxmertForPreTrainingOutput
7474
:members:
7575

76-
.. autoclass:: transformers.modeling_lxmert.LxmertForQuestionAnsweringOutput
76+
.. autoclass:: transformers.models.lxmert.modeling_lxmert.LxmertForQuestionAnsweringOutput
7777
:members:
7878

79-
.. autoclass:: transformers.modeling_tf_lxmert.TFLxmertModelOutput
79+
.. autoclass:: transformers.models.lxmert.modeling_tf_lxmert.TFLxmertModelOutput
8080
:members:
8181

82-
.. autoclass:: transformers.modeling_tf_lxmert.TFLxmertForPreTrainingOutput
82+
.. autoclass:: transformers.models.lxmert.modeling_tf_lxmert.TFLxmertForPreTrainingOutput
8383
:members:
8484

8585

‎docs/source/model_doc/mobilebert.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,10 +58,10 @@ MobileBertTokenizerFast
5858
MobileBert specific outputs
5959
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
6060

61-
.. autoclass:: transformers.modeling_mobilebert.MobileBertForPreTrainingOutput
61+
.. autoclass:: transformers.models.mobilebert.modeling_mobilebert.MobileBertForPreTrainingOutput
6262
:members:
6363

64-
.. autoclass:: transformers.modeling_tf_mobilebert.TFMobileBertForPreTrainingOutput
64+
.. autoclass:: transformers.models.mobilebert.modeling_tf_mobilebert.TFMobileBertForPreTrainingOutput
6565
:members:
6666

6767

‎docs/source/model_doc/prophetnet.rst

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,16 +47,16 @@ ProphetNetTokenizer
4747
ProphetNet specific outputs
4848
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
4949

50-
.. autoclass:: transformers.modeling_prophetnet.ProphetNetSeq2SeqLMOutput
50+
.. autoclass:: transformers.models.prophetnet.modeling_prophetnet.ProphetNetSeq2SeqLMOutput
5151
:members:
5252

53-
.. autoclass:: transformers.modeling_prophetnet.ProphetNetSeq2SeqModelOutput
53+
.. autoclass:: transformers.models.prophetnet.modeling_prophetnet.ProphetNetSeq2SeqModelOutput
5454
:members:
5555

56-
.. autoclass:: transformers.modeling_prophetnet.ProphetNetDecoderModelOutput
56+
.. autoclass:: transformers.models.prophetnet.modeling_prophetnet.ProphetNetDecoderModelOutput
5757
:members:
5858

59-
.. autoclass:: transformers.modeling_prophetnet.ProphetNetDecoderLMOutput
59+
.. autoclass:: transformers.models.prophetnet.modeling_prophetnet.ProphetNetDecoderLMOutput
6060
:members:
6161

6262
ProphetNetModel

‎docs/source/model_doc/rag.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,10 +50,10 @@ RagTokenizer
5050
Rag specific outputs
5151
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
5252

53-
.. autoclass:: transformers.modeling_rag.RetrievAugLMMarginOutput
53+
.. autoclass:: transformers.models.rag.modeling_rag.RetrievAugLMMarginOutput
5454
:members:
5555

56-
.. autoclass:: transformers.modeling_rag.RetrievAugLMOutput
56+
.. autoclass:: transformers.models.rag.modeling_rag.RetrievAugLMOutput
5757
:members:
5858

5959
RagRetriever

‎docs/source/model_doc/transformerxl.rst

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -49,16 +49,16 @@ TransfoXLTokenizer
4949
TransfoXL specific outputs
5050
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
5151

52-
.. autoclass:: transformers.modeling_transfo_xl.TransfoXLModelOutput
52+
.. autoclass:: transformers.models.transfo_xl.modeling_transfo_xl.TransfoXLModelOutput
5353
:members:
5454

55-
.. autoclass:: transformers.modeling_transfo_xl.TransfoXLLMHeadModelOutput
55+
.. autoclass:: transformers.models.transfo_xl.modeling_transfo_xl.TransfoXLLMHeadModelOutput
5656
:members:
5757

58-
.. autoclass:: transformers.modeling_tf_transfo_xl.TFTransfoXLModelOutput
58+
.. autoclass:: transformers.models.transfo_xl.modeling_tf_transfo_xl.TFTransfoXLModelOutput
5959
:members:
6060

61-
.. autoclass:: transformers.modeling_tf_transfo_xl.TFTransfoXLLMHeadModelOutput
61+
.. autoclass:: transformers.models.transfo_xl.modeling_tf_transfo_xl.TFTransfoXLLMHeadModelOutput
6262
:members:
6363

6464

‎docs/source/model_doc/xlm.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ XLMTokenizer
5050
XLM specific outputs
5151
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
5252

53-
.. autoclass:: transformers.modeling_xlm.XLMForQuestionAnsweringOutput
53+
.. autoclass:: transformers.models.xlm.modeling_xlm.XLMForQuestionAnsweringOutput
5454
:members:
5555

5656

‎docs/source/model_doc/xlnet.rst

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -53,43 +53,43 @@ XLNetTokenizer
5353
XLNet specific outputs
5454
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
5555

56-
.. autoclass:: transformers.modeling_xlnet.XLNetModelOutput
56+
.. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetModelOutput
5757
:members:
5858

59-
.. autoclass:: transformers.modeling_xlnet.XLNetLMHeadModelOutput
59+
.. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetLMHeadModelOutput
6060
:members:
6161

62-
.. autoclass:: transformers.modeling_xlnet.XLNetForSequenceClassificationOutput
62+
.. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForSequenceClassificationOutput
6363
:members:
6464

65-
.. autoclass:: transformers.modeling_xlnet.XLNetForMultipleChoiceOutput
65+
.. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForMultipleChoiceOutput
6666
:members:
6767

68-
.. autoclass:: transformers.modeling_xlnet.XLNetForTokenClassificationOutput
68+
.. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForTokenClassificationOutput
6969
:members:
7070

71-
.. autoclass:: transformers.modeling_xlnet.XLNetForQuestionAnsweringSimpleOutput
71+
.. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForQuestionAnsweringSimpleOutput
7272
:members:
7373

74-
.. autoclass:: transformers.modeling_xlnet.XLNetForQuestionAnsweringOutput
74+
.. autoclass:: transformers.models.xlnet.modeling_xlnet.XLNetForQuestionAnsweringOutput
7575
:members:
7676

77-
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetModelOutput
77+
.. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetModelOutput
7878
:members:
7979

80-
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetLMHeadModelOutput
80+
.. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetLMHeadModelOutput
8181
:members:
8282

83-
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetForSequenceClassificationOutput
83+
.. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetForSequenceClassificationOutput
8484
:members:
8585

86-
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetForMultipleChoiceOutput
86+
.. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetForMultipleChoiceOutput
8787
:members:
8888

89-
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetForTokenClassificationOutput
89+
.. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetForTokenClassificationOutput
9090
:members:
9191

92-
.. autoclass:: transformers.modeling_tf_xlnet.TFXLNetForQuestionAnsweringSimpleOutput
92+
.. autoclass:: transformers.models.xlnet.modeling_tf_xlnet.TFXLNetForQuestionAnsweringSimpleOutput
9393
:members:
9494

9595

‎examples/bert-loses-patience/pabee/modeling_pabee_albert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from torch.nn import CrossEntropyLoss, MSELoss
2222

2323
from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_model_forward
24-
from transformers.modeling_albert import (
24+
from transformers.models.albert.modeling_albert import (
2525
ALBERT_INPUTS_DOCSTRING,
2626
ALBERT_START_DOCSTRING,
2727
AlbertModel,

‎examples/bert-loses-patience/pabee/modeling_pabee_bert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from torch.nn import CrossEntropyLoss, MSELoss
2424

2525
from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_model_forward
26-
from transformers.modeling_bert import (
26+
from transformers.models.bert.modeling_bert import (
2727
BERT_INPUTS_DOCSTRING,
2828
BERT_START_DOCSTRING,
2929
BertEncoder,

‎examples/contrib/run_camembert.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import torch
22

3-
from transformers.modeling_camembert import CamembertForMaskedLM
4-
from transformers.tokenization_camembert import CamembertTokenizer
3+
from transformers import CamembertForMaskedLM, CamembertTokenizer
54

65

76
def fill_mask(masked_input, model, tokenizer, topk=5):

‎examples/contrib/run_swag.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,14 @@
3232
from tqdm import tqdm, trange
3333

3434
import transformers
35-
from transformers import WEIGHTS_NAME, AdamW, AutoConfig, AutoTokenizer, get_linear_schedule_with_warmup
36-
from transformers.modeling_auto import AutoModelForMultipleChoice
35+
from transformers import (
36+
WEIGHTS_NAME,
37+
AdamW,
38+
AutoConfig,
39+
AutoModelForMultipleChoice,
40+
AutoTokenizer,
41+
get_linear_schedule_with_warmup,
42+
)
3743
from transformers.trainer_utils import is_main_process
3844

3945

‎examples/deebert/src/modeling_highway_bert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from torch.nn import CrossEntropyLoss, MSELoss
44

55
from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_model_forward
6-
from transformers.modeling_bert import (
6+
from transformers.models.bert.modeling_bert import (
77
BERT_INPUTS_DOCSTRING,
88
BERT_START_DOCSTRING,
99
BertEmbeddings,

‎examples/deebert/src/modeling_highway_roberta.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,13 @@
33
import torch.nn as nn
44
from torch.nn import CrossEntropyLoss, MSELoss
55

6-
from transformers.configuration_roberta import RobertaConfig
6+
from transformers import RobertaConfig
77
from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_model_forward
8-
from transformers.modeling_roberta import ROBERTA_INPUTS_DOCSTRING, ROBERTA_START_DOCSTRING, RobertaEmbeddings
8+
from transformers.models.roberta.modeling_roberta import (
9+
ROBERTA_INPUTS_DOCSTRING,
10+
ROBERTA_START_DOCSTRING,
11+
RobertaEmbeddings,
12+
)
913

1014
from .modeling_highway_bert import BertPreTrainedModel, DeeBertModel, HighwayException, entropy
1115

0 commit comments

Comments
 (0)