dummy_flax_objects.py 5.84 KB
Newer Older
1
# This file is autogenerated by the command `make fix-copies`, do not edit.
2
from ..file_utils import requires_backends
3
4


5
6
class FlaxPreTrainedModel:
    def __init__(self, *args, **kwargs):
7
        requires_backends(self, ["flax"])
8
9
10

    @classmethod
    def from_pretrained(self, *args, **kwargs):
11
        requires_backends(self, ["flax"])
12
13


14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
FLAX_MODEL_FOR_MASKED_LM_MAPPING = None


FLAX_MODEL_FOR_MULTIPLE_CHOICE_MAPPING = None


FLAX_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING = None


FLAX_MODEL_FOR_PRETRAINING_MAPPING = None


FLAX_MODEL_FOR_QUESTION_ANSWERING_MAPPING = None


FLAX_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = None


FLAX_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING = None


Sylvain Gugger's avatar
Sylvain Gugger committed
35
36
37
38
39
FLAX_MODEL_MAPPING = None


class FlaxAutoModel:
    def __init__(self, *args, **kwargs):
40
        requires_backends(self, ["flax"])
Sylvain Gugger's avatar
Sylvain Gugger committed
41
42
43

    @classmethod
    def from_pretrained(self, *args, **kwargs):
44
        requires_backends(self, ["flax"])
Sylvain Gugger's avatar
Sylvain Gugger committed
45
46


47
48
class FlaxAutoModelForMaskedLM:
    def __init__(self, *args, **kwargs):
49
        requires_backends(self, ["flax"])
50
51
52

    @classmethod
    def from_pretrained(self, *args, **kwargs):
53
        requires_backends(self, ["flax"])
54
55
56
57


class FlaxAutoModelForMultipleChoice:
    def __init__(self, *args, **kwargs):
58
        requires_backends(self, ["flax"])
59
60
61

    @classmethod
    def from_pretrained(self, *args, **kwargs):
62
        requires_backends(self, ["flax"])
63
64
65
66


class FlaxAutoModelForNextSentencePrediction:
    def __init__(self, *args, **kwargs):
67
        requires_backends(self, ["flax"])
68
69
70

    @classmethod
    def from_pretrained(self, *args, **kwargs):
71
        requires_backends(self, ["flax"])
72
73
74
75


class FlaxAutoModelForPreTraining:
    def __init__(self, *args, **kwargs):
76
        requires_backends(self, ["flax"])
77
78
79

    @classmethod
    def from_pretrained(self, *args, **kwargs):
80
        requires_backends(self, ["flax"])
81
82
83
84


class FlaxAutoModelForQuestionAnswering:
    def __init__(self, *args, **kwargs):
85
        requires_backends(self, ["flax"])
86
87
88

    @classmethod
    def from_pretrained(self, *args, **kwargs):
89
        requires_backends(self, ["flax"])
90
91
92
93


class FlaxAutoModelForSequenceClassification:
    def __init__(self, *args, **kwargs):
94
        requires_backends(self, ["flax"])
95
96
97

    @classmethod
    def from_pretrained(self, *args, **kwargs):
98
        requires_backends(self, ["flax"])
99
100
101
102


class FlaxAutoModelForTokenClassification:
    def __init__(self, *args, **kwargs):
103
        requires_backends(self, ["flax"])
104
105
106

    @classmethod
    def from_pretrained(self, *args, **kwargs):
107
        requires_backends(self, ["flax"])
108
109


110
111
class FlaxBertForMaskedLM:
    def __init__(self, *args, **kwargs):
112
        requires_backends(self, ["flax"])
113
114
115

    @classmethod
    def from_pretrained(self, *args, **kwargs):
116
        requires_backends(self, ["flax"])
117
118


119
120
class FlaxBertForMultipleChoice:
    def __init__(self, *args, **kwargs):
121
        requires_backends(self, ["flax"])
122
123
124

    @classmethod
    def from_pretrained(self, *args, **kwargs):
125
        requires_backends(self, ["flax"])
126
127
128
129


class FlaxBertForNextSentencePrediction:
    def __init__(self, *args, **kwargs):
130
        requires_backends(self, ["flax"])
131
132
133
134


class FlaxBertForPreTraining:
    def __init__(self, *args, **kwargs):
135
        requires_backends(self, ["flax"])
136
137
138
139


class FlaxBertForQuestionAnswering:
    def __init__(self, *args, **kwargs):
140
        requires_backends(self, ["flax"])
141
142
143

    @classmethod
    def from_pretrained(self, *args, **kwargs):
144
        requires_backends(self, ["flax"])
145
146
147
148


class FlaxBertForSequenceClassification:
    def __init__(self, *args, **kwargs):
149
        requires_backends(self, ["flax"])
150
151
152

    @classmethod
    def from_pretrained(self, *args, **kwargs):
153
        requires_backends(self, ["flax"])
154
155
156
157


class FlaxBertForTokenClassification:
    def __init__(self, *args, **kwargs):
158
        requires_backends(self, ["flax"])
159
160
161

    @classmethod
    def from_pretrained(self, *args, **kwargs):
162
        requires_backends(self, ["flax"])
163
164


165
166
class FlaxBertModel:
    def __init__(self, *args, **kwargs):
167
        requires_backends(self, ["flax"])
168
169
170

    @classmethod
    def from_pretrained(self, *args, **kwargs):
171
        requires_backends(self, ["flax"])
172
173


174
175
class FlaxBertPreTrainedModel:
    def __init__(self, *args, **kwargs):
176
        requires_backends(self, ["flax"])
177
178
179

    @classmethod
    def from_pretrained(self, *args, **kwargs):
180
        requires_backends(self, ["flax"])
181
182


183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
class FlaxRobertaForMaskedLM:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_backends(self, ["flax"])


class FlaxRobertaForMultipleChoice:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_backends(self, ["flax"])


class FlaxRobertaForQuestionAnswering:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_backends(self, ["flax"])


class FlaxRobertaForSequenceClassification:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_backends(self, ["flax"])


class FlaxRobertaForTokenClassification:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_backends(self, ["flax"])


228
229
class FlaxRobertaModel:
    def __init__(self, *args, **kwargs):
230
        requires_backends(self, ["flax"])
231
232
233

    @classmethod
    def from_pretrained(self, *args, **kwargs):
234
        requires_backends(self, ["flax"])
235
236
237
238
239
240
241
242
243


class FlaxRobertaPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_backends(self, ["flax"])