"vscode:/vscode.git/clone" did not exist on "3f6add8bab8245f819a8950b7c5d09aff9366e32"
dummy_flax_objects.py 9.73 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


Patrick von Platen's avatar
Patrick von Platen committed
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
class FlaxLogitsProcessor:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])


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


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


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


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


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


35
36
class FlaxPreTrainedModel:
    def __init__(self, *args, **kwargs):
37
        requires_backends(self, ["flax"])
38
39
40

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


Suraj Patil's avatar
Suraj Patil committed
44
45
46
FLAX_MODEL_FOR_CAUSAL_LM_MAPPING = None


47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
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
68
69
70
71
72
FLAX_MODEL_MAPPING = None


class FlaxAutoModel:
    def __init__(self, *args, **kwargs):
73
        requires_backends(self, ["flax"])
Sylvain Gugger's avatar
Sylvain Gugger committed
74
75
76

    @classmethod
    def from_pretrained(self, *args, **kwargs):
77
        requires_backends(self, ["flax"])
Sylvain Gugger's avatar
Sylvain Gugger committed
78
79


Suraj Patil's avatar
Suraj Patil committed
80
81
82
83
84
85
86
87
88
class FlaxAutoModelForCausalLM:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

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


89
90
class FlaxAutoModelForMaskedLM:
    def __init__(self, *args, **kwargs):
91
        requires_backends(self, ["flax"])
92
93
94

    @classmethod
    def from_pretrained(self, *args, **kwargs):
95
        requires_backends(self, ["flax"])
96
97
98
99


class FlaxAutoModelForMultipleChoice:
    def __init__(self, *args, **kwargs):
100
        requires_backends(self, ["flax"])
101
102
103

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


class FlaxAutoModelForNextSentencePrediction:
    def __init__(self, *args, **kwargs):
109
        requires_backends(self, ["flax"])
110
111
112

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


class FlaxAutoModelForPreTraining:
    def __init__(self, *args, **kwargs):
118
        requires_backends(self, ["flax"])
119
120
121

    @classmethod
    def from_pretrained(self, *args, **kwargs):
122
        requires_backends(self, ["flax"])
123
124
125
126


class FlaxAutoModelForQuestionAnswering:
    def __init__(self, *args, **kwargs):
127
        requires_backends(self, ["flax"])
128
129
130

    @classmethod
    def from_pretrained(self, *args, **kwargs):
131
        requires_backends(self, ["flax"])
132
133
134
135


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

    @classmethod
    def from_pretrained(self, *args, **kwargs):
140
        requires_backends(self, ["flax"])
141
142
143
144


class FlaxAutoModelForTokenClassification:
    def __init__(self, *args, **kwargs):
145
        requires_backends(self, ["flax"])
146
147
148

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


152
153
class FlaxBertForMaskedLM:
    def __init__(self, *args, **kwargs):
154
        requires_backends(self, ["flax"])
155
156
157

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


161
162
class FlaxBertForMultipleChoice:
    def __init__(self, *args, **kwargs):
163
        requires_backends(self, ["flax"])
164
165
166

    @classmethod
    def from_pretrained(self, *args, **kwargs):
167
        requires_backends(self, ["flax"])
168
169
170
171


class FlaxBertForNextSentencePrediction:
    def __init__(self, *args, **kwargs):
172
        requires_backends(self, ["flax"])
173
174
175
176


class FlaxBertForPreTraining:
    def __init__(self, *args, **kwargs):
177
        requires_backends(self, ["flax"])
178
179
180
181


class FlaxBertForQuestionAnswering:
    def __init__(self, *args, **kwargs):
182
        requires_backends(self, ["flax"])
183
184
185

    @classmethod
    def from_pretrained(self, *args, **kwargs):
186
        requires_backends(self, ["flax"])
187
188
189
190


class FlaxBertForSequenceClassification:
    def __init__(self, *args, **kwargs):
191
        requires_backends(self, ["flax"])
192
193
194

    @classmethod
    def from_pretrained(self, *args, **kwargs):
195
        requires_backends(self, ["flax"])
196
197
198
199


class FlaxBertForTokenClassification:
    def __init__(self, *args, **kwargs):
200
        requires_backends(self, ["flax"])
201
202
203

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


207
208
class FlaxBertModel:
    def __init__(self, *args, **kwargs):
209
        requires_backends(self, ["flax"])
210
211
212

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


216
217
class FlaxBertPreTrainedModel:
    def __init__(self, *args, **kwargs):
218
        requires_backends(self, ["flax"])
219
220
221

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


Suraj Patil's avatar
Suraj Patil committed
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
class FlaxCLIPModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

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


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

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


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

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


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

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


261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
class FlaxElectraForMaskedLM:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

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


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

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


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


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

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


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

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


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

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


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

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


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

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


Suraj Patil's avatar
Suraj Patil committed
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
class FlaxGPT2LMHeadModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["flax"])

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


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

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


347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
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"])


392
393
class FlaxRobertaModel:
    def __init__(self, *args, **kwargs):
394
        requires_backends(self, ["flax"])
395
396
397

    @classmethod
    def from_pretrained(self, *args, **kwargs):
398
        requires_backends(self, ["flax"])
399
400
401
402
403
404
405
406
407


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

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