"doc/vscode:/vscode.git/clone" did not exist on "09f5c5e77e81a5a0c352457b5687c6d9973fda6c"
dummy_tf_objects.py 45.8 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 TensorFlowBenchmarkArguments:
    def __init__(self, *args, **kwargs):
7
        requires_backends(self, ["tf"])
8
9
10
11


class TensorFlowBenchmark:
    def __init__(self, *args, **kwargs):
12
        requires_backends(self, ["tf"])
13
14
15


def tf_top_k_top_p_filtering(*args, **kwargs):
16
    requires_backends(tf_top_k_top_p_filtering, ["tf"])
17
18


Amir Tahmasbi's avatar
Amir Tahmasbi committed
19
20
21
22
23
TF_LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFLayoutLMForMaskedLM:
    def __init__(self, *args, **kwargs):
24
        requires_backends(self, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
25
26

    @classmethod
27
28
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
29
30
31
32


class TFLayoutLMForSequenceClassification:
    def __init__(self, *args, **kwargs):
33
        requires_backends(self, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
34
35

    @classmethod
36
37
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
38
39
40
41


class TFLayoutLMForTokenClassification:
    def __init__(self, *args, **kwargs):
42
        requires_backends(self, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
43
44

    @classmethod
45
46
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
47
48
49
50


class TFLayoutLMMainLayer:
    def __init__(self, *args, **kwargs):
51
        requires_backends(self, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
52
53
54
55


class TFLayoutLMModel:
    def __init__(self, *args, **kwargs):
56
        requires_backends(self, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
57
58

    @classmethod
59
60
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
61
62
63
64


class TFLayoutLMPreTrainedModel:
    def __init__(self, *args, **kwargs):
65
        requires_backends(self, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
66
67

    @classmethod
68
69
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Amir Tahmasbi's avatar
Amir Tahmasbi committed
70
71


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

    @classmethod
77
78
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Sylvain Gugger's avatar
Sylvain Gugger committed
79
80
81
82


class TFSequenceSummary:
    def __init__(self, *args, **kwargs):
83
        requires_backends(self, ["tf"])
Sylvain Gugger's avatar
Sylvain Gugger committed
84
85
86
87


class TFSharedEmbeddings:
    def __init__(self, *args, **kwargs):
88
        requires_backends(self, ["tf"])
Sylvain Gugger's avatar
Sylvain Gugger committed
89
90
91


def shape_list(*args, **kwargs):
92
    requires_backends(shape_list, ["tf"])
Sylvain Gugger's avatar
Sylvain Gugger committed
93
94


95
96
97
98
99
TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFAlbertForMaskedLM:
    def __init__(self, *args, **kwargs):
100
        requires_backends(self, ["tf"])
101
102

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


class TFAlbertForMultipleChoice:
    def __init__(self, *args, **kwargs):
109
        requires_backends(self, ["tf"])
110
111

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


class TFAlbertForPreTraining:
    def __init__(self, *args, **kwargs):
118
        requires_backends(self, ["tf"])
119
120
121
122


class TFAlbertForQuestionAnswering:
    def __init__(self, *args, **kwargs):
123
        requires_backends(self, ["tf"])
124
125

    @classmethod
126
127
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
128
129
130
131


class TFAlbertForSequenceClassification:
    def __init__(self, *args, **kwargs):
132
        requires_backends(self, ["tf"])
133
134

    @classmethod
135
136
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
137
138
139
140


class TFAlbertForTokenClassification:
    def __init__(self, *args, **kwargs):
141
        requires_backends(self, ["tf"])
142
143

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


class TFAlbertMainLayer:
    def __init__(self, *args, **kwargs):
150
        requires_backends(self, ["tf"])
151
152
153
154


class TFAlbertModel:
    def __init__(self, *args, **kwargs):
155
        requires_backends(self, ["tf"])
156
157

    @classmethod
158
159
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
160
161
162
163


class TFAlbertPreTrainedModel:
    def __init__(self, *args, **kwargs):
164
        requires_backends(self, ["tf"])
165
166

    @classmethod
167
168
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
169
170
171
172
173
174
175
176
177
178
179


TF_MODEL_FOR_CAUSAL_LM_MAPPING = None


TF_MODEL_FOR_MASKED_LM_MAPPING = None


TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING = None


Julien Plu's avatar
Julien Plu committed
180
181
182
TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING = None


183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
TF_MODEL_FOR_PRETRAINING_MAPPING = None


TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING = None


TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING = None


TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = None


TF_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING = None


TF_MODEL_MAPPING = None


TF_MODEL_WITH_LM_HEAD_MAPPING = None


class TFAutoModel:
    def __init__(self, *args, **kwargs):
206
        requires_backends(self, ["tf"])
207
208

    @classmethod
209
210
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
211
212
213
214


class TFAutoModelForCausalLM:
    def __init__(self, *args, **kwargs):
215
        requires_backends(self, ["tf"])
216
217

    @classmethod
218
219
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
220
221
222
223


class TFAutoModelForMaskedLM:
    def __init__(self, *args, **kwargs):
224
        requires_backends(self, ["tf"])
225
226

    @classmethod
227
228
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
229
230
231
232


class TFAutoModelForMultipleChoice:
    def __init__(self, *args, **kwargs):
233
        requires_backends(self, ["tf"])
234
235

    @classmethod
236
237
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
238
239
240
241


class TFAutoModelForPreTraining:
    def __init__(self, *args, **kwargs):
242
        requires_backends(self, ["tf"])
243
244

    @classmethod
245
246
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
247
248
249
250


class TFAutoModelForQuestionAnswering:
    def __init__(self, *args, **kwargs):
251
        requires_backends(self, ["tf"])
252
253

    @classmethod
254
255
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
256
257
258
259


class TFAutoModelForSeq2SeqLM:
    def __init__(self, *args, **kwargs):
260
        requires_backends(self, ["tf"])
261
262

    @classmethod
263
264
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
265
266
267
268


class TFAutoModelForSequenceClassification:
    def __init__(self, *args, **kwargs):
269
        requires_backends(self, ["tf"])
270
271

    @classmethod
272
273
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
274
275
276
277


class TFAutoModelForTokenClassification:
    def __init__(self, *args, **kwargs):
278
        requires_backends(self, ["tf"])
279
280

    @classmethod
281
282
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
283
284
285
286


class TFAutoModelWithLMHead:
    def __init__(self, *args, **kwargs):
287
        requires_backends(self, ["tf"])
288
289

    @classmethod
290
291
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
292
293


294
295
class TFBartForConditionalGeneration:
    def __init__(self, *args, **kwargs):
296
        requires_backends(self, ["tf"])
297
298

    @classmethod
299
300
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
301
302
303
304


class TFBartModel:
    def __init__(self, *args, **kwargs):
305
        requires_backends(self, ["tf"])
306
307

    @classmethod
308
309
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
310
311


312
313
class TFBartPretrainedModel:
    def __init__(self, *args, **kwargs):
314
        requires_backends(self, ["tf"])
315
316

    @classmethod
317
318
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
319
320


321
322
323
324
325
TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFBertEmbeddings:
    def __init__(self, *args, **kwargs):
326
        requires_backends(self, ["tf"])
327
328
329
330


class TFBertForMaskedLM:
    def __init__(self, *args, **kwargs):
331
        requires_backends(self, ["tf"])
332
333

    @classmethod
334
335
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
336
337
338
339


class TFBertForMultipleChoice:
    def __init__(self, *args, **kwargs):
340
        requires_backends(self, ["tf"])
341
342

    @classmethod
343
344
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
345
346
347
348


class TFBertForNextSentencePrediction:
    def __init__(self, *args, **kwargs):
349
        requires_backends(self, ["tf"])
350
351
352
353


class TFBertForPreTraining:
    def __init__(self, *args, **kwargs):
354
        requires_backends(self, ["tf"])
355
356
357
358


class TFBertForQuestionAnswering:
    def __init__(self, *args, **kwargs):
359
        requires_backends(self, ["tf"])
360
361

    @classmethod
362
363
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
364
365
366
367


class TFBertForSequenceClassification:
    def __init__(self, *args, **kwargs):
368
        requires_backends(self, ["tf"])
369
370

    @classmethod
371
372
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
373
374
375
376


class TFBertForTokenClassification:
    def __init__(self, *args, **kwargs):
377
        requires_backends(self, ["tf"])
378
379

    @classmethod
380
381
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
382
383
384
385


class TFBertLMHeadModel:
    def __init__(self, *args, **kwargs):
386
        requires_backends(self, ["tf"])
387
388

    @classmethod
389
390
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
391
392
393
394


class TFBertMainLayer:
    def __init__(self, *args, **kwargs):
395
        requires_backends(self, ["tf"])
396
397
398
399


class TFBertModel:
    def __init__(self, *args, **kwargs):
400
        requires_backends(self, ["tf"])
401
402

    @classmethod
403
404
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
405
406
407
408


class TFBertPreTrainedModel:
    def __init__(self, *args, **kwargs):
409
        requires_backends(self, ["tf"])
410
411

    @classmethod
412
413
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
414
415


416
417
class TFBlenderbotForConditionalGeneration:
    def __init__(self, *args, **kwargs):
418
        requires_backends(self, ["tf"])
419
420

    @classmethod
421
422
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
423
424


425
426
class TFBlenderbotModel:
    def __init__(self, *args, **kwargs):
427
        requires_backends(self, ["tf"])
428
429

    @classmethod
430
431
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
432
433


434
435
436
437
438
439
440
441
442
class TFBlenderbotPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


443
444
class TFBlenderbotSmallForConditionalGeneration:
    def __init__(self, *args, **kwargs):
445
        requires_backends(self, ["tf"])
446
447

    @classmethod
448
449
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
450
451
452
453


class TFBlenderbotSmallModel:
    def __init__(self, *args, **kwargs):
454
        requires_backends(self, ["tf"])
455
456

    @classmethod
457
458
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
459
460


461
462
463
464
465
466
467
468
469
class TFBlenderbotSmallPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


470
471
472
473
474
TF_CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFCamembertForMaskedLM:
    def __init__(self, *args, **kwargs):
475
        requires_backends(self, ["tf"])
476
477

    @classmethod
478
479
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
480
481
482
483


class TFCamembertForMultipleChoice:
    def __init__(self, *args, **kwargs):
484
        requires_backends(self, ["tf"])
485
486

    @classmethod
487
488
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
489
490
491
492


class TFCamembertForQuestionAnswering:
    def __init__(self, *args, **kwargs):
493
        requires_backends(self, ["tf"])
494
495

    @classmethod
496
497
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
498
499
500
501


class TFCamembertForSequenceClassification:
    def __init__(self, *args, **kwargs):
502
        requires_backends(self, ["tf"])
503
504

    @classmethod
505
506
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
507
508
509
510


class TFCamembertForTokenClassification:
    def __init__(self, *args, **kwargs):
511
        requires_backends(self, ["tf"])
512
513

    @classmethod
514
515
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
516
517
518
519


class TFCamembertModel:
    def __init__(self, *args, **kwargs):
520
        requires_backends(self, ["tf"])
521
522

    @classmethod
523
524
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
525
526


abhishek thakur's avatar
abhishek thakur committed
527
528
529
530
531
TF_CONVBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFConvBertForMaskedLM:
    def __init__(self, *args, **kwargs):
532
        requires_backends(self, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
533
534

    @classmethod
535
536
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
537
538
539
540


class TFConvBertForMultipleChoice:
    def __init__(self, *args, **kwargs):
541
        requires_backends(self, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
542
543

    @classmethod
544
545
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
546
547
548
549


class TFConvBertForQuestionAnswering:
    def __init__(self, *args, **kwargs):
550
        requires_backends(self, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
551
552

    @classmethod
553
554
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
555
556
557
558


class TFConvBertForSequenceClassification:
    def __init__(self, *args, **kwargs):
559
        requires_backends(self, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
560
561

    @classmethod
562
563
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
564
565
566
567


class TFConvBertForTokenClassification:
    def __init__(self, *args, **kwargs):
568
        requires_backends(self, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
569
570

    @classmethod
571
572
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
573
574
575
576


class TFConvBertLayer:
    def __init__(self, *args, **kwargs):
577
        requires_backends(self, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
578
579
580
581


class TFConvBertModel:
    def __init__(self, *args, **kwargs):
582
        requires_backends(self, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
583
584

    @classmethod
585
586
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
587
588
589
590


class TFConvBertPreTrainedModel:
    def __init__(self, *args, **kwargs):
591
        requires_backends(self, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
592
593

    @classmethod
594
595
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
abhishek thakur's avatar
abhishek thakur committed
596
597


598
599
600
TF_CTRL_PRETRAINED_MODEL_ARCHIVE_LIST = None


601
602
class TFCTRLForSequenceClassification:
    def __init__(self, *args, **kwargs):
603
        requires_backends(self, ["tf"])
604
605

    @classmethod
606
607
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
608
609


610
611
class TFCTRLLMHeadModel:
    def __init__(self, *args, **kwargs):
612
        requires_backends(self, ["tf"])
613
614

    @classmethod
615
616
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
617
618
619
620


class TFCTRLModel:
    def __init__(self, *args, **kwargs):
621
        requires_backends(self, ["tf"])
622
623

    @classmethod
624
625
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
626
627
628
629


class TFCTRLPreTrainedModel:
    def __init__(self, *args, **kwargs):
630
        requires_backends(self, ["tf"])
631
632

    @classmethod
633
634
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
635
636
637
638
639
640
641


TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFDistilBertForMaskedLM:
    def __init__(self, *args, **kwargs):
642
        requires_backends(self, ["tf"])
643
644

    @classmethod
645
646
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
647
648
649
650


class TFDistilBertForMultipleChoice:
    def __init__(self, *args, **kwargs):
651
        requires_backends(self, ["tf"])
652
653

    @classmethod
654
655
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
656
657
658
659


class TFDistilBertForQuestionAnswering:
    def __init__(self, *args, **kwargs):
660
        requires_backends(self, ["tf"])
661
662

    @classmethod
663
664
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
665
666
667
668


class TFDistilBertForSequenceClassification:
    def __init__(self, *args, **kwargs):
669
        requires_backends(self, ["tf"])
670
671

    @classmethod
672
673
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
674
675
676
677


class TFDistilBertForTokenClassification:
    def __init__(self, *args, **kwargs):
678
        requires_backends(self, ["tf"])
679
680

    @classmethod
681
682
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
683
684
685
686


class TFDistilBertMainLayer:
    def __init__(self, *args, **kwargs):
687
        requires_backends(self, ["tf"])
688
689
690
691


class TFDistilBertModel:
    def __init__(self, *args, **kwargs):
692
        requires_backends(self, ["tf"])
693
694

    @classmethod
695
696
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
697
698
699
700


class TFDistilBertPreTrainedModel:
    def __init__(self, *args, **kwargs):
701
        requires_backends(self, ["tf"])
702
703

    @classmethod
704
705
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
706
707


Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
708
709
710
711
712
713
714
715
716
717
718
TF_DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None


TF_DPR_QUESTION_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None


TF_DPR_READER_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFDPRContextEncoder:
    def __init__(self, *args, **kwargs):
719
        requires_backends(self, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
720
721
722
723


class TFDPRPretrainedContextEncoder:
    def __init__(self, *args, **kwargs):
724
        requires_backends(self, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
725
726
727
728


class TFDPRPretrainedQuestionEncoder:
    def __init__(self, *args, **kwargs):
729
        requires_backends(self, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
730
731
732
733


class TFDPRPretrainedReader:
    def __init__(self, *args, **kwargs):
734
        requires_backends(self, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
735
736
737
738


class TFDPRQuestionEncoder:
    def __init__(self, *args, **kwargs):
739
        requires_backends(self, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
740
741
742
743


class TFDPRReader:
    def __init__(self, *args, **kwargs):
744
        requires_backends(self, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
745
746


747
748
749
750
751
TF_ELECTRA_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFElectraForMaskedLM:
    def __init__(self, *args, **kwargs):
752
        requires_backends(self, ["tf"])
753
754

    @classmethod
755
756
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
757
758
759
760


class TFElectraForMultipleChoice:
    def __init__(self, *args, **kwargs):
761
        requires_backends(self, ["tf"])
762
763

    @classmethod
764
765
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
766
767
768
769


class TFElectraForPreTraining:
    def __init__(self, *args, **kwargs):
770
        requires_backends(self, ["tf"])
771
772
773
774


class TFElectraForQuestionAnswering:
    def __init__(self, *args, **kwargs):
775
        requires_backends(self, ["tf"])
776
777

    @classmethod
778
779
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
780
781
782
783


class TFElectraForSequenceClassification:
    def __init__(self, *args, **kwargs):
784
        requires_backends(self, ["tf"])
785
786

    @classmethod
787
788
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
789
790
791
792


class TFElectraForTokenClassification:
    def __init__(self, *args, **kwargs):
793
        requires_backends(self, ["tf"])
794
795

    @classmethod
796
797
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
798
799
800
801


class TFElectraModel:
    def __init__(self, *args, **kwargs):
802
        requires_backends(self, ["tf"])
803
804

    @classmethod
805
806
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
807
808
809
810


class TFElectraPreTrainedModel:
    def __init__(self, *args, **kwargs):
811
        requires_backends(self, ["tf"])
812
813

    @classmethod
814
815
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
816
817
818
819
820
821
822


TF_FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFFlaubertForMultipleChoice:
    def __init__(self, *args, **kwargs):
823
        requires_backends(self, ["tf"])
824
825

    @classmethod
826
827
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
828
829
830
831


class TFFlaubertForQuestionAnsweringSimple:
    def __init__(self, *args, **kwargs):
832
        requires_backends(self, ["tf"])
833
834

    @classmethod
835
836
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
837
838
839
840


class TFFlaubertForSequenceClassification:
    def __init__(self, *args, **kwargs):
841
        requires_backends(self, ["tf"])
842
843

    @classmethod
844
845
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
846
847
848
849


class TFFlaubertForTokenClassification:
    def __init__(self, *args, **kwargs):
850
        requires_backends(self, ["tf"])
851
852

    @classmethod
853
854
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
855
856
857
858


class TFFlaubertModel:
    def __init__(self, *args, **kwargs):
859
        requires_backends(self, ["tf"])
860
861

    @classmethod
862
863
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
864
865


866
867
868
869
870
871
872
873
874
class TFFlaubertPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


875
876
class TFFlaubertWithLMHeadModel:
    def __init__(self, *args, **kwargs):
877
        requires_backends(self, ["tf"])
878
879

    @classmethod
880
881
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
882
883
884
885
886
887
888


TF_FUNNEL_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFFunnelBaseModel:
    def __init__(self, *args, **kwargs):
889
        requires_backends(self, ["tf"])
890
891

    @classmethod
892
893
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
894
895
896
897


class TFFunnelForMaskedLM:
    def __init__(self, *args, **kwargs):
898
        requires_backends(self, ["tf"])
899
900

    @classmethod
901
902
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
903
904
905
906


class TFFunnelForMultipleChoice:
    def __init__(self, *args, **kwargs):
907
        requires_backends(self, ["tf"])
908
909

    @classmethod
910
911
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
912
913
914
915


class TFFunnelForPreTraining:
    def __init__(self, *args, **kwargs):
916
        requires_backends(self, ["tf"])
917
918
919
920


class TFFunnelForQuestionAnswering:
    def __init__(self, *args, **kwargs):
921
        requires_backends(self, ["tf"])
922
923

    @classmethod
924
925
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
926
927
928
929


class TFFunnelForSequenceClassification:
    def __init__(self, *args, **kwargs):
930
        requires_backends(self, ["tf"])
931
932

    @classmethod
933
934
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
935
936
937
938


class TFFunnelForTokenClassification:
    def __init__(self, *args, **kwargs):
939
        requires_backends(self, ["tf"])
940
941

    @classmethod
942
943
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
944
945
946
947


class TFFunnelModel:
    def __init__(self, *args, **kwargs):
948
        requires_backends(self, ["tf"])
949
950

    @classmethod
951
952
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
953
954


955
956
957
958
959
960
961
962
963
class TFFunnelPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


964
965
966
967
968
TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFGPT2DoubleHeadsModel:
    def __init__(self, *args, **kwargs):
969
        requires_backends(self, ["tf"])
970
971

    @classmethod
972
973
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
974
975
976
977


class TFGPT2ForSequenceClassification:
    def __init__(self, *args, **kwargs):
978
        requires_backends(self, ["tf"])
979
980

    @classmethod
981
982
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
983
984
985
986


class TFGPT2LMHeadModel:
    def __init__(self, *args, **kwargs):
987
        requires_backends(self, ["tf"])
988
989

    @classmethod
990
991
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
992
993
994
995


class TFGPT2MainLayer:
    def __init__(self, *args, **kwargs):
996
        requires_backends(self, ["tf"])
997
998
999
1000


class TFGPT2Model:
    def __init__(self, *args, **kwargs):
1001
        requires_backends(self, ["tf"])
1002
1003

    @classmethod
1004
1005
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1006
1007
1008
1009


class TFGPT2PreTrainedModel:
    def __init__(self, *args, **kwargs):
1010
        requires_backends(self, ["tf"])
1011
1012

    @classmethod
1013
1014
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1015
1016


Patrick von Platen's avatar
Patrick von Platen committed
1017
1018
class TFLEDForConditionalGeneration:
    def __init__(self, *args, **kwargs):
1019
        requires_backends(self, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1020
1021

    @classmethod
1022
1023
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1024
1025
1026
1027


class TFLEDModel:
    def __init__(self, *args, **kwargs):
1028
        requires_backends(self, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1029
1030

    @classmethod
1031
1032
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1033
1034
1035
1036


class TFLEDPreTrainedModel:
    def __init__(self, *args, **kwargs):
1037
        requires_backends(self, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1038
1039

    @classmethod
1040
1041
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1042
1043


1044
1045
1046
1047
1048
TF_LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFLongformerForMaskedLM:
    def __init__(self, *args, **kwargs):
1049
        requires_backends(self, ["tf"])
1050
1051

    @classmethod
1052
1053
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1054
1055


1056
1057
class TFLongformerForMultipleChoice:
    def __init__(self, *args, **kwargs):
1058
        requires_backends(self, ["tf"])
1059
1060

    @classmethod
1061
1062
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1063
1064


1065
1066
class TFLongformerForQuestionAnswering:
    def __init__(self, *args, **kwargs):
1067
        requires_backends(self, ["tf"])
1068
1069

    @classmethod
1070
1071
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1072
1073
1074
1075


class TFLongformerForSequenceClassification:
    def __init__(self, *args, **kwargs):
1076
        requires_backends(self, ["tf"])
1077
1078

    @classmethod
1079
1080
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1081
1082
1083
1084


class TFLongformerForTokenClassification:
    def __init__(self, *args, **kwargs):
1085
        requires_backends(self, ["tf"])
1086
1087

    @classmethod
1088
1089
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1090
1091
1092
1093


class TFLongformerModel:
    def __init__(self, *args, **kwargs):
1094
        requires_backends(self, ["tf"])
1095
1096

    @classmethod
1097
1098
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1099
1100


1101
1102
1103
1104
1105
1106
1107
1108
1109
class TFLongformerPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


1110
1111
class TFLongformerSelfAttention:
    def __init__(self, *args, **kwargs):
1112
        requires_backends(self, ["tf"])
1113
1114
1115
1116
1117
1118
1119


TF_LXMERT_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFLxmertForPreTraining:
    def __init__(self, *args, **kwargs):
1120
        requires_backends(self, ["tf"])
1121
1122
1123
1124


class TFLxmertMainLayer:
    def __init__(self, *args, **kwargs):
1125
        requires_backends(self, ["tf"])
1126
1127
1128
1129


class TFLxmertModel:
    def __init__(self, *args, **kwargs):
1130
        requires_backends(self, ["tf"])
1131
1132

    @classmethod
1133
1134
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1135
1136
1137
1138


class TFLxmertPreTrainedModel:
    def __init__(self, *args, **kwargs):
1139
        requires_backends(self, ["tf"])
1140
1141

    @classmethod
1142
1143
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1144
1145
1146
1147


class TFLxmertVisualFeatureEncoder:
    def __init__(self, *args, **kwargs):
1148
        requires_backends(self, ["tf"])
1149
1150


Sylvain Gugger's avatar
Sylvain Gugger committed
1151
class TFMarianModel:
1152
    def __init__(self, *args, **kwargs):
1153
        requires_backends(self, ["tf"])
1154

Sylvain Gugger's avatar
Sylvain Gugger committed
1155
    @classmethod
1156
1157
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Sylvain Gugger's avatar
Sylvain Gugger committed
1158

1159

1160
1161
class TFMarianMTModel:
    def __init__(self, *args, **kwargs):
1162
        requires_backends(self, ["tf"])
1163
1164

    @classmethod
1165
1166
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1167
1168


1169
1170
1171
1172
1173
1174
1175
1176
1177
class TFMarianPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


1178
1179
class TFMBartForConditionalGeneration:
    def __init__(self, *args, **kwargs):
1180
        requires_backends(self, ["tf"])
1181
1182

    @classmethod
1183
1184
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1185
1186


1187
1188
class TFMBartModel:
    def __init__(self, *args, **kwargs):
1189
        requires_backends(self, ["tf"])
1190
1191

    @classmethod
1192
1193
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1194
1195


1196
1197
1198
1199
1200
1201
1202
1203
1204
class TFMBartPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


1205
1206
1207
1208
1209
TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFMobileBertForMaskedLM:
    def __init__(self, *args, **kwargs):
1210
        requires_backends(self, ["tf"])
1211
1212

    @classmethod
1213
1214
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1215
1216
1217
1218


class TFMobileBertForMultipleChoice:
    def __init__(self, *args, **kwargs):
1219
        requires_backends(self, ["tf"])
1220
1221

    @classmethod
1222
1223
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1224
1225
1226
1227


class TFMobileBertForNextSentencePrediction:
    def __init__(self, *args, **kwargs):
1228
        requires_backends(self, ["tf"])
1229
1230
1231
1232


class TFMobileBertForPreTraining:
    def __init__(self, *args, **kwargs):
1233
        requires_backends(self, ["tf"])
1234
1235
1236
1237


class TFMobileBertForQuestionAnswering:
    def __init__(self, *args, **kwargs):
1238
        requires_backends(self, ["tf"])
1239
1240

    @classmethod
1241
1242
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1243
1244
1245
1246


class TFMobileBertForSequenceClassification:
    def __init__(self, *args, **kwargs):
1247
        requires_backends(self, ["tf"])
1248
1249

    @classmethod
1250
1251
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1252
1253
1254
1255


class TFMobileBertForTokenClassification:
    def __init__(self, *args, **kwargs):
1256
        requires_backends(self, ["tf"])
1257
1258

    @classmethod
1259
1260
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1261
1262
1263
1264


class TFMobileBertMainLayer:
    def __init__(self, *args, **kwargs):
1265
        requires_backends(self, ["tf"])
1266
1267
1268
1269


class TFMobileBertModel:
    def __init__(self, *args, **kwargs):
1270
        requires_backends(self, ["tf"])
1271
1272

    @classmethod
1273
1274
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1275
1276
1277
1278


class TFMobileBertPreTrainedModel:
    def __init__(self, *args, **kwargs):
1279
        requires_backends(self, ["tf"])
1280
1281

    @classmethod
1282
1283
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1284
1285


StillKeepTry's avatar
StillKeepTry committed
1286
1287
1288
1289
1290
TF_MPNET_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFMPNetForMaskedLM:
    def __init__(self, *args, **kwargs):
1291
        requires_backends(self, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1292
1293

    @classmethod
1294
1295
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1296
1297
1298
1299


class TFMPNetForMultipleChoice:
    def __init__(self, *args, **kwargs):
1300
        requires_backends(self, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1301
1302

    @classmethod
1303
1304
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1305
1306
1307
1308


class TFMPNetForQuestionAnswering:
    def __init__(self, *args, **kwargs):
1309
        requires_backends(self, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1310
1311

    @classmethod
1312
1313
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1314
1315
1316
1317


class TFMPNetForSequenceClassification:
    def __init__(self, *args, **kwargs):
1318
        requires_backends(self, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1319
1320

    @classmethod
1321
1322
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1323
1324
1325
1326


class TFMPNetForTokenClassification:
    def __init__(self, *args, **kwargs):
1327
        requires_backends(self, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1328
1329

    @classmethod
1330
1331
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1332
1333
1334
1335


class TFMPNetMainLayer:
    def __init__(self, *args, **kwargs):
1336
        requires_backends(self, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1337
1338
1339
1340


class TFMPNetModel:
    def __init__(self, *args, **kwargs):
1341
        requires_backends(self, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1342
1343

    @classmethod
1344
1345
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1346
1347
1348
1349


class TFMPNetPreTrainedModel:
    def __init__(self, *args, **kwargs):
1350
        requires_backends(self, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1351
1352

    @classmethod
1353
1354
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
StillKeepTry's avatar
StillKeepTry committed
1355
1356


1357
1358
class TFMT5EncoderModel:
    def __init__(self, *args, **kwargs):
1359
        requires_backends(self, ["tf"])
1360
1361

    @classmethod
1362
1363
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1364
1365


Patrick von Platen's avatar
Patrick von Platen committed
1366
1367
class TFMT5ForConditionalGeneration:
    def __init__(self, *args, **kwargs):
1368
        requires_backends(self, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1369
1370

    @classmethod
1371
1372
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1373
1374
1375
1376


class TFMT5Model:
    def __init__(self, *args, **kwargs):
1377
        requires_backends(self, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1378
1379

    @classmethod
1380
1381
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Patrick von Platen's avatar
Patrick von Platen committed
1382
1383


1384
1385
1386
1387
1388
TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFOpenAIGPTDoubleHeadsModel:
    def __init__(self, *args, **kwargs):
1389
        requires_backends(self, ["tf"])
1390
1391

    @classmethod
1392
1393
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1394
1395


1396
1397
class TFOpenAIGPTForSequenceClassification:
    def __init__(self, *args, **kwargs):
1398
        requires_backends(self, ["tf"])
1399
1400

    @classmethod
1401
1402
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1403
1404


1405
1406
class TFOpenAIGPTLMHeadModel:
    def __init__(self, *args, **kwargs):
1407
        requires_backends(self, ["tf"])
1408
1409

    @classmethod
1410
1411
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1412
1413
1414
1415


class TFOpenAIGPTMainLayer:
    def __init__(self, *args, **kwargs):
1416
        requires_backends(self, ["tf"])
1417
1418
1419
1420


class TFOpenAIGPTModel:
    def __init__(self, *args, **kwargs):
1421
        requires_backends(self, ["tf"])
1422
1423

    @classmethod
1424
1425
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1426
1427
1428
1429


class TFOpenAIGPTPreTrainedModel:
    def __init__(self, *args, **kwargs):
1430
        requires_backends(self, ["tf"])
1431
1432

    @classmethod
1433
1434
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1435
1436


1437
1438
class TFPegasusForConditionalGeneration:
    def __init__(self, *args, **kwargs):
1439
        requires_backends(self, ["tf"])
1440
1441

    @classmethod
1442
1443
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1444
1445


1446
1447
class TFPegasusModel:
    def __init__(self, *args, **kwargs):
1448
        requires_backends(self, ["tf"])
1449
1450

    @classmethod
1451
1452
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1453
1454


1455
1456
1457
1458
1459
1460
1461
1462
1463
class TFPegasusPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
1464
1465
class TFRagModel:
    def __init__(self, *args, **kwargs):
1466
        requires_backends(self, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
1467
1468

    @classmethod
1469
1470
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
1471
1472


1473
1474
1475
1476
1477
1478
1479
1480
1481
class TFRagPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
1482
1483
class TFRagSequenceForGeneration:
    def __init__(self, *args, **kwargs):
1484
        requires_backends(self, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
1485
1486
1487
1488


class TFRagTokenForGeneration:
    def __init__(self, *args, **kwargs):
1489
        requires_backends(self, ["tf"])
Ratthachat (Jung)'s avatar
Ratthachat (Jung) committed
1490
1491


1492
1493
1494
1495
1496
TF_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFRobertaForMaskedLM:
    def __init__(self, *args, **kwargs):
1497
        requires_backends(self, ["tf"])
1498
1499

    @classmethod
1500
1501
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1502
1503
1504
1505


class TFRobertaForMultipleChoice:
    def __init__(self, *args, **kwargs):
1506
        requires_backends(self, ["tf"])
1507
1508

    @classmethod
1509
1510
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1511
1512
1513
1514


class TFRobertaForQuestionAnswering:
    def __init__(self, *args, **kwargs):
1515
        requires_backends(self, ["tf"])
1516
1517

    @classmethod
1518
1519
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1520
1521
1522
1523


class TFRobertaForSequenceClassification:
    def __init__(self, *args, **kwargs):
1524
        requires_backends(self, ["tf"])
1525
1526

    @classmethod
1527
1528
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1529
1530
1531
1532


class TFRobertaForTokenClassification:
    def __init__(self, *args, **kwargs):
1533
        requires_backends(self, ["tf"])
1534
1535

    @classmethod
1536
1537
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1538
1539
1540
1541


class TFRobertaMainLayer:
    def __init__(self, *args, **kwargs):
1542
        requires_backends(self, ["tf"])
1543
1544
1545
1546


class TFRobertaModel:
    def __init__(self, *args, **kwargs):
1547
        requires_backends(self, ["tf"])
1548
1549

    @classmethod
1550
1551
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1552
1553
1554
1555


class TFRobertaPreTrainedModel:
    def __init__(self, *args, **kwargs):
1556
        requires_backends(self, ["tf"])
1557
1558

    @classmethod
1559
1560
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1561
1562


1563
1564
1565
1566
1567
1568
1569
TF_ROFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFRoFormerForCausalLM:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

1570
1571
1572
1573
    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])

1574
1575
1576
1577
1578
1579

class TFRoFormerForMaskedLM:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
1580
1581
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1582
1583
1584
1585
1586
1587
1588


class TFRoFormerForMultipleChoice:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
1589
1590
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1591
1592
1593
1594
1595
1596
1597


class TFRoFormerForQuestionAnswering:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
1598
1599
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1600
1601
1602
1603
1604
1605
1606


class TFRoFormerForSequenceClassification:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
1607
1608
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1609
1610
1611
1612
1613
1614
1615


class TFRoFormerForTokenClassification:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
1616
1617
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1618
1619
1620
1621
1622
1623
1624
1625
1626
1627
1628
1629


class TFRoFormerLayer:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])


class TFRoFormerModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
1630
1631
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1632
1633
1634
1635
1636
1637
1638


class TFRoFormerPreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
1639
1640
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1641
1642


1643
1644
1645
TF_T5_PRETRAINED_MODEL_ARCHIVE_LIST = None


1646
1647
class TFT5EncoderModel:
    def __init__(self, *args, **kwargs):
1648
        requires_backends(self, ["tf"])
1649
1650

    @classmethod
1651
1652
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1653
1654


1655
1656
class TFT5ForConditionalGeneration:
    def __init__(self, *args, **kwargs):
1657
        requires_backends(self, ["tf"])
1658
1659

    @classmethod
1660
1661
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1662
1663
1664
1665


class TFT5Model:
    def __init__(self, *args, **kwargs):
1666
        requires_backends(self, ["tf"])
1667
1668

    @classmethod
1669
1670
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1671
1672
1673
1674


class TFT5PreTrainedModel:
    def __init__(self, *args, **kwargs):
1675
        requires_backends(self, ["tf"])
1676
1677

    @classmethod
1678
1679
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1680
1681
1682
1683
1684
1685
1686


TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFAdaptiveEmbedding:
    def __init__(self, *args, **kwargs):
1687
        requires_backends(self, ["tf"])
1688
1689


1690
1691
class TFTransfoXLForSequenceClassification:
    def __init__(self, *args, **kwargs):
1692
        requires_backends(self, ["tf"])
1693
1694

    @classmethod
1695
1696
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1697
1698


1699
1700
class TFTransfoXLLMHeadModel:
    def __init__(self, *args, **kwargs):
1701
        requires_backends(self, ["tf"])
1702
1703

    @classmethod
1704
1705
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1706
1707
1708
1709


class TFTransfoXLMainLayer:
    def __init__(self, *args, **kwargs):
1710
        requires_backends(self, ["tf"])
1711
1712
1713
1714


class TFTransfoXLModel:
    def __init__(self, *args, **kwargs):
1715
        requires_backends(self, ["tf"])
1716
1717

    @classmethod
1718
1719
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1720
1721
1722
1723


class TFTransfoXLPreTrainedModel:
    def __init__(self, *args, **kwargs):
1724
        requires_backends(self, ["tf"])
1725
1726

    @classmethod
1727
1728
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1729
1730


Will Rice's avatar
Will Rice committed
1731
1732
1733
1734
1735
1736
1737
1738
1739
1740
1741
1742
1743
1744
1745
1746
1747
1748
1749
1750
1751
1752
1753
1754
1755
1756
TF_WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFWav2Vec2ForCTC:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])


class TFWav2Vec2Model:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


class TFWav2Vec2PreTrainedModel:
    def __init__(self, *args, **kwargs):
        requires_backends(self, ["tf"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])


1757
1758
1759
1760
1761
TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFXLMForMultipleChoice:
    def __init__(self, *args, **kwargs):
1762
        requires_backends(self, ["tf"])
1763
1764

    @classmethod
1765
1766
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1767
1768
1769
1770


class TFXLMForQuestionAnsweringSimple:
    def __init__(self, *args, **kwargs):
1771
        requires_backends(self, ["tf"])
1772
1773

    @classmethod
1774
1775
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1776
1777
1778
1779


class TFXLMForSequenceClassification:
    def __init__(self, *args, **kwargs):
1780
        requires_backends(self, ["tf"])
1781
1782

    @classmethod
1783
1784
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1785
1786
1787
1788


class TFXLMForTokenClassification:
    def __init__(self, *args, **kwargs):
1789
        requires_backends(self, ["tf"])
1790
1791

    @classmethod
1792
1793
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1794
1795
1796
1797


class TFXLMMainLayer:
    def __init__(self, *args, **kwargs):
1798
        requires_backends(self, ["tf"])
1799
1800
1801
1802


class TFXLMModel:
    def __init__(self, *args, **kwargs):
1803
        requires_backends(self, ["tf"])
1804
1805

    @classmethod
1806
1807
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1808
1809
1810
1811


class TFXLMPreTrainedModel:
    def __init__(self, *args, **kwargs):
1812
        requires_backends(self, ["tf"])
1813
1814

    @classmethod
1815
1816
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1817
1818
1819
1820


class TFXLMWithLMHeadModel:
    def __init__(self, *args, **kwargs):
1821
        requires_backends(self, ["tf"])
1822
1823

    @classmethod
1824
1825
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1826
1827
1828
1829
1830
1831
1832


TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFXLMRobertaForMaskedLM:
    def __init__(self, *args, **kwargs):
1833
        requires_backends(self, ["tf"])
1834
1835

    @classmethod
1836
1837
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1838
1839
1840
1841


class TFXLMRobertaForMultipleChoice:
    def __init__(self, *args, **kwargs):
1842
        requires_backends(self, ["tf"])
1843
1844

    @classmethod
1845
1846
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1847
1848
1849
1850


class TFXLMRobertaForQuestionAnswering:
    def __init__(self, *args, **kwargs):
1851
        requires_backends(self, ["tf"])
1852
1853

    @classmethod
1854
1855
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1856
1857
1858
1859


class TFXLMRobertaForSequenceClassification:
    def __init__(self, *args, **kwargs):
1860
        requires_backends(self, ["tf"])
1861
1862

    @classmethod
1863
1864
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1865
1866
1867
1868


class TFXLMRobertaForTokenClassification:
    def __init__(self, *args, **kwargs):
1869
        requires_backends(self, ["tf"])
1870
1871

    @classmethod
1872
1873
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1874
1875
1876
1877


class TFXLMRobertaModel:
    def __init__(self, *args, **kwargs):
1878
        requires_backends(self, ["tf"])
1879
1880

    @classmethod
1881
1882
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1883
1884
1885
1886
1887
1888
1889


TF_XLNET_PRETRAINED_MODEL_ARCHIVE_LIST = None


class TFXLNetForMultipleChoice:
    def __init__(self, *args, **kwargs):
1890
        requires_backends(self, ["tf"])
1891
1892

    @classmethod
1893
1894
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1895
1896
1897
1898


class TFXLNetForQuestionAnsweringSimple:
    def __init__(self, *args, **kwargs):
1899
        requires_backends(self, ["tf"])
1900
1901

    @classmethod
1902
1903
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1904
1905
1906
1907


class TFXLNetForSequenceClassification:
    def __init__(self, *args, **kwargs):
1908
        requires_backends(self, ["tf"])
1909
1910

    @classmethod
1911
1912
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1913
1914
1915
1916


class TFXLNetForTokenClassification:
    def __init__(self, *args, **kwargs):
1917
        requires_backends(self, ["tf"])
1918
1919

    @classmethod
1920
1921
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1922
1923
1924
1925


class TFXLNetLMHeadModel:
    def __init__(self, *args, **kwargs):
1926
        requires_backends(self, ["tf"])
1927
1928

    @classmethod
1929
1930
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1931
1932
1933
1934


class TFXLNetMainLayer:
    def __init__(self, *args, **kwargs):
1935
        requires_backends(self, ["tf"])
1936
1937
1938
1939


class TFXLNetModel:
    def __init__(self, *args, **kwargs):
1940
        requires_backends(self, ["tf"])
1941
1942

    @classmethod
1943
1944
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1945
1946
1947
1948


class TFXLNetPreTrainedModel:
    def __init__(self, *args, **kwargs):
1949
        requires_backends(self, ["tf"])
1950
1951

    @classmethod
1952
1953
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, ["tf"])
1954
1955
1956
1957


class AdamWeightDecay:
    def __init__(self, *args, **kwargs):
1958
        requires_backends(self, ["tf"])
1959
1960
1961
1962


class GradientAccumulator:
    def __init__(self, *args, **kwargs):
1963
        requires_backends(self, ["tf"])
1964
1965
1966
1967


class WarmUp:
    def __init__(self, *args, **kwargs):
1968
        requires_backends(self, ["tf"])
1969
1970
1971


def create_optimizer(*args, **kwargs):
1972
    requires_backends(create_optimizer, ["tf"])
1973
1974
1975
1976


class TFTrainer:
    def __init__(self, *args, **kwargs):
1977
        requires_backends(self, ["tf"])