Skip to content

GitLab

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

Add forward method to dummy models (#14419) 

* Add forward method to dummy models

* Fix quality
parent 040fd471
Show whitespace changes
Inline Side-by-side
Showing with 2071 additions and 4 deletions
src/transformers/utils/dummy_flax_objects.py
View file @ 3e8d17e6
...@@ -75,6 +75,9 @@ class FlaxPreTrainedModel: ...@@ -75,6 +75,9 @@ class FlaxPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAlbertForMaskedLM: class FlaxAlbertForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -84,6 +87,9 @@ class FlaxAlbertForMaskedLM: ...@@ -84,6 +87,9 @@ class FlaxAlbertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAlbertForMultipleChoice: class FlaxAlbertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -93,6 +99,9 @@ class FlaxAlbertForMultipleChoice: ...@@ -93,6 +99,9 @@ class FlaxAlbertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAlbertForPreTraining: class FlaxAlbertForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -107,6 +116,9 @@ class FlaxAlbertForQuestionAnswering: ...@@ -107,6 +116,9 @@ class FlaxAlbertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAlbertForSequenceClassification: class FlaxAlbertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -116,6 +128,9 @@ class FlaxAlbertForSequenceClassification: ...@@ -116,6 +128,9 @@ class FlaxAlbertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAlbertForTokenClassification: class FlaxAlbertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -125,6 +140,9 @@ class FlaxAlbertForTokenClassification: ...@@ -125,6 +140,9 @@ class FlaxAlbertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAlbertModel: class FlaxAlbertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -134,6 +152,9 @@ class FlaxAlbertModel: ...@@ -134,6 +152,9 @@ class FlaxAlbertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAlbertPreTrainedModel: class FlaxAlbertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -143,6 +164,9 @@ class FlaxAlbertPreTrainedModel: ...@@ -143,6 +164,9 @@ class FlaxAlbertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
FLAX_MODEL_FOR_CAUSAL_LM_MAPPING = None FLAX_MODEL_FOR_CAUSAL_LM_MAPPING = None
...@@ -188,6 +212,9 @@ class FlaxAutoModel: ...@@ -188,6 +212,9 @@ class FlaxAutoModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForCausalLM: class FlaxAutoModelForCausalLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -197,6 +224,9 @@ class FlaxAutoModelForCausalLM: ...@@ -197,6 +224,9 @@ class FlaxAutoModelForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForImageClassification: class FlaxAutoModelForImageClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -206,6 +236,9 @@ class FlaxAutoModelForImageClassification: ...@@ -206,6 +236,9 @@ class FlaxAutoModelForImageClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForMaskedLM: class FlaxAutoModelForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -215,6 +248,9 @@ class FlaxAutoModelForMaskedLM: ...@@ -215,6 +248,9 @@ class FlaxAutoModelForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForMultipleChoice: class FlaxAutoModelForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -224,6 +260,9 @@ class FlaxAutoModelForMultipleChoice: ...@@ -224,6 +260,9 @@ class FlaxAutoModelForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForNextSentencePrediction: class FlaxAutoModelForNextSentencePrediction:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -233,6 +272,9 @@ class FlaxAutoModelForNextSentencePrediction: ...@@ -233,6 +272,9 @@ class FlaxAutoModelForNextSentencePrediction:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForPreTraining: class FlaxAutoModelForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -242,6 +284,9 @@ class FlaxAutoModelForPreTraining: ...@@ -242,6 +284,9 @@ class FlaxAutoModelForPreTraining:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForQuestionAnswering: class FlaxAutoModelForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -251,6 +296,9 @@ class FlaxAutoModelForQuestionAnswering: ...@@ -251,6 +296,9 @@ class FlaxAutoModelForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForSeq2SeqLM: class FlaxAutoModelForSeq2SeqLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -260,6 +308,9 @@ class FlaxAutoModelForSeq2SeqLM: ...@@ -260,6 +308,9 @@ class FlaxAutoModelForSeq2SeqLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForSequenceClassification: class FlaxAutoModelForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -269,6 +320,9 @@ class FlaxAutoModelForSequenceClassification: ...@@ -269,6 +320,9 @@ class FlaxAutoModelForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForTokenClassification: class FlaxAutoModelForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -278,6 +332,9 @@ class FlaxAutoModelForTokenClassification: ...@@ -278,6 +332,9 @@ class FlaxAutoModelForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxAutoModelForVision2Seq: class FlaxAutoModelForVision2Seq:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -287,6 +344,9 @@ class FlaxAutoModelForVision2Seq: ...@@ -287,6 +344,9 @@ class FlaxAutoModelForVision2Seq:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBartForConditionalGeneration: class FlaxBartForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -296,6 +356,9 @@ class FlaxBartForConditionalGeneration: ...@@ -296,6 +356,9 @@ class FlaxBartForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBartForQuestionAnswering: class FlaxBartForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -305,6 +368,9 @@ class FlaxBartForQuestionAnswering: ...@@ -305,6 +368,9 @@ class FlaxBartForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBartForSequenceClassification: class FlaxBartForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -314,6 +380,9 @@ class FlaxBartForSequenceClassification: ...@@ -314,6 +380,9 @@ class FlaxBartForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBartModel: class FlaxBartModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -323,6 +392,9 @@ class FlaxBartModel: ...@@ -323,6 +392,9 @@ class FlaxBartModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBartPreTrainedModel: class FlaxBartPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -332,6 +404,9 @@ class FlaxBartPreTrainedModel: ...@@ -332,6 +404,9 @@ class FlaxBartPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBeitForImageClassification: class FlaxBeitForImageClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -346,6 +421,9 @@ class FlaxBeitForMaskedImageModeling: ...@@ -346,6 +421,9 @@ class FlaxBeitForMaskedImageModeling:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBeitModel: class FlaxBeitModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -355,6 +433,9 @@ class FlaxBeitModel: ...@@ -355,6 +433,9 @@ class FlaxBeitModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBeitPreTrainedModel: class FlaxBeitPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -364,6 +445,9 @@ class FlaxBeitPreTrainedModel: ...@@ -364,6 +445,9 @@ class FlaxBeitPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBertForMaskedLM: class FlaxBertForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -373,6 +457,9 @@ class FlaxBertForMaskedLM: ...@@ -373,6 +457,9 @@ class FlaxBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBertForMultipleChoice: class FlaxBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -382,6 +469,9 @@ class FlaxBertForMultipleChoice: ...@@ -382,6 +469,9 @@ class FlaxBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBertForNextSentencePrediction: class FlaxBertForNextSentencePrediction:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -401,6 +491,9 @@ class FlaxBertForQuestionAnswering: ...@@ -401,6 +491,9 @@ class FlaxBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBertForSequenceClassification: class FlaxBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -410,6 +503,9 @@ class FlaxBertForSequenceClassification: ...@@ -410,6 +503,9 @@ class FlaxBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBertForTokenClassification: class FlaxBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -419,6 +515,9 @@ class FlaxBertForTokenClassification: ...@@ -419,6 +515,9 @@ class FlaxBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBertModel: class FlaxBertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -428,6 +527,9 @@ class FlaxBertModel: ...@@ -428,6 +527,9 @@ class FlaxBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBertPreTrainedModel: class FlaxBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -437,6 +539,9 @@ class FlaxBertPreTrainedModel: ...@@ -437,6 +539,9 @@ class FlaxBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBigBirdForMaskedLM: class FlaxBigBirdForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -446,6 +551,9 @@ class FlaxBigBirdForMaskedLM: ...@@ -446,6 +551,9 @@ class FlaxBigBirdForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBigBirdForMultipleChoice: class FlaxBigBirdForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -455,6 +563,9 @@ class FlaxBigBirdForMultipleChoice: ...@@ -455,6 +563,9 @@ class FlaxBigBirdForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBigBirdForPreTraining: class FlaxBigBirdForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -469,6 +580,9 @@ class FlaxBigBirdForQuestionAnswering: ...@@ -469,6 +580,9 @@ class FlaxBigBirdForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBigBirdForSequenceClassification: class FlaxBigBirdForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -478,6 +592,9 @@ class FlaxBigBirdForSequenceClassification: ...@@ -478,6 +592,9 @@ class FlaxBigBirdForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBigBirdForTokenClassification: class FlaxBigBirdForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -487,6 +604,9 @@ class FlaxBigBirdForTokenClassification: ...@@ -487,6 +604,9 @@ class FlaxBigBirdForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBigBirdModel: class FlaxBigBirdModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -496,6 +616,9 @@ class FlaxBigBirdModel: ...@@ -496,6 +616,9 @@ class FlaxBigBirdModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxBigBirdPreTrainedModel: class FlaxBigBirdPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -505,6 +628,9 @@ class FlaxBigBirdPreTrainedModel: ...@@ -505,6 +628,9 @@ class FlaxBigBirdPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxCLIPModel: class FlaxCLIPModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -514,6 +640,9 @@ class FlaxCLIPModel: ...@@ -514,6 +640,9 @@ class FlaxCLIPModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxCLIPPreTrainedModel: class FlaxCLIPPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -523,6 +652,9 @@ class FlaxCLIPPreTrainedModel: ...@@ -523,6 +652,9 @@ class FlaxCLIPPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxCLIPTextModel: class FlaxCLIPTextModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -532,6 +664,9 @@ class FlaxCLIPTextModel: ...@@ -532,6 +664,9 @@ class FlaxCLIPTextModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxCLIPTextPreTrainedModel: class FlaxCLIPTextPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -541,6 +676,9 @@ class FlaxCLIPTextPreTrainedModel: ...@@ -541,6 +676,9 @@ class FlaxCLIPTextPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxCLIPVisionModel: class FlaxCLIPVisionModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -550,6 +688,9 @@ class FlaxCLIPVisionModel: ...@@ -550,6 +688,9 @@ class FlaxCLIPVisionModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxCLIPVisionPreTrainedModel: class FlaxCLIPVisionPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -559,6 +700,9 @@ class FlaxCLIPVisionPreTrainedModel: ...@@ -559,6 +700,9 @@ class FlaxCLIPVisionPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxDistilBertForMaskedLM: class FlaxDistilBertForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -568,6 +712,9 @@ class FlaxDistilBertForMaskedLM: ...@@ -568,6 +712,9 @@ class FlaxDistilBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxDistilBertForMultipleChoice: class FlaxDistilBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -577,6 +724,9 @@ class FlaxDistilBertForMultipleChoice: ...@@ -577,6 +724,9 @@ class FlaxDistilBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxDistilBertForQuestionAnswering: class FlaxDistilBertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -586,6 +736,9 @@ class FlaxDistilBertForQuestionAnswering: ...@@ -586,6 +736,9 @@ class FlaxDistilBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxDistilBertForSequenceClassification: class FlaxDistilBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -595,6 +748,9 @@ class FlaxDistilBertForSequenceClassification: ...@@ -595,6 +748,9 @@ class FlaxDistilBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxDistilBertForTokenClassification: class FlaxDistilBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -604,6 +760,9 @@ class FlaxDistilBertForTokenClassification: ...@@ -604,6 +760,9 @@ class FlaxDistilBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxDistilBertModel: class FlaxDistilBertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -613,6 +772,9 @@ class FlaxDistilBertModel: ...@@ -613,6 +772,9 @@ class FlaxDistilBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxDistilBertPreTrainedModel: class FlaxDistilBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -622,6 +784,9 @@ class FlaxDistilBertPreTrainedModel: ...@@ -622,6 +784,9 @@ class FlaxDistilBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxElectraForMaskedLM: class FlaxElectraForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -631,6 +796,9 @@ class FlaxElectraForMaskedLM: ...@@ -631,6 +796,9 @@ class FlaxElectraForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxElectraForMultipleChoice: class FlaxElectraForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -640,6 +808,9 @@ class FlaxElectraForMultipleChoice: ...@@ -640,6 +808,9 @@ class FlaxElectraForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxElectraForPreTraining: class FlaxElectraForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -654,6 +825,9 @@ class FlaxElectraForQuestionAnswering: ...@@ -654,6 +825,9 @@ class FlaxElectraForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxElectraForSequenceClassification: class FlaxElectraForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -663,6 +837,9 @@ class FlaxElectraForSequenceClassification: ...@@ -663,6 +837,9 @@ class FlaxElectraForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxElectraForTokenClassification: class FlaxElectraForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -672,6 +849,9 @@ class FlaxElectraForTokenClassification: ...@@ -672,6 +849,9 @@ class FlaxElectraForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxElectraModel: class FlaxElectraModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -681,6 +861,9 @@ class FlaxElectraModel: ...@@ -681,6 +861,9 @@ class FlaxElectraModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxElectraPreTrainedModel: class FlaxElectraPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -690,6 +873,9 @@ class FlaxElectraPreTrainedModel: ...@@ -690,6 +873,9 @@ class FlaxElectraPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxEncoderDecoderModel: class FlaxEncoderDecoderModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -699,6 +885,9 @@ class FlaxEncoderDecoderModel: ...@@ -699,6 +885,9 @@ class FlaxEncoderDecoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxGPT2LMHeadModel: class FlaxGPT2LMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -708,6 +897,9 @@ class FlaxGPT2LMHeadModel: ...@@ -708,6 +897,9 @@ class FlaxGPT2LMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxGPT2Model: class FlaxGPT2Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -717,6 +909,9 @@ class FlaxGPT2Model: ...@@ -717,6 +909,9 @@ class FlaxGPT2Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxGPT2PreTrainedModel: class FlaxGPT2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -726,6 +921,9 @@ class FlaxGPT2PreTrainedModel: ...@@ -726,6 +921,9 @@ class FlaxGPT2PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxGPTNeoForCausalLM: class FlaxGPTNeoForCausalLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -735,6 +933,9 @@ class FlaxGPTNeoForCausalLM: ...@@ -735,6 +933,9 @@ class FlaxGPTNeoForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxGPTNeoModel: class FlaxGPTNeoModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -744,6 +945,9 @@ class FlaxGPTNeoModel: ...@@ -744,6 +945,9 @@ class FlaxGPTNeoModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxGPTNeoPreTrainedModel: class FlaxGPTNeoPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -753,6 +957,9 @@ class FlaxGPTNeoPreTrainedModel: ...@@ -753,6 +957,9 @@ class FlaxGPTNeoPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMarianModel: class FlaxMarianModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -762,6 +969,9 @@ class FlaxMarianModel: ...@@ -762,6 +969,9 @@ class FlaxMarianModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMarianMTModel: class FlaxMarianMTModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -771,6 +981,9 @@ class FlaxMarianMTModel: ...@@ -771,6 +981,9 @@ class FlaxMarianMTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMarianPreTrainedModel: class FlaxMarianPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -780,6 +993,9 @@ class FlaxMarianPreTrainedModel: ...@@ -780,6 +993,9 @@ class FlaxMarianPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMBartForConditionalGeneration: class FlaxMBartForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -789,6 +1005,9 @@ class FlaxMBartForConditionalGeneration: ...@@ -789,6 +1005,9 @@ class FlaxMBartForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMBartForQuestionAnswering: class FlaxMBartForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -798,6 +1017,9 @@ class FlaxMBartForQuestionAnswering: ...@@ -798,6 +1017,9 @@ class FlaxMBartForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMBartForSequenceClassification: class FlaxMBartForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -807,6 +1029,9 @@ class FlaxMBartForSequenceClassification: ...@@ -807,6 +1029,9 @@ class FlaxMBartForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMBartModel: class FlaxMBartModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -816,6 +1041,9 @@ class FlaxMBartModel: ...@@ -816,6 +1041,9 @@ class FlaxMBartModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMBartPreTrainedModel: class FlaxMBartPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -825,6 +1053,9 @@ class FlaxMBartPreTrainedModel: ...@@ -825,6 +1053,9 @@ class FlaxMBartPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMT5ForConditionalGeneration: class FlaxMT5ForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -834,6 +1065,9 @@ class FlaxMT5ForConditionalGeneration: ...@@ -834,6 +1065,9 @@ class FlaxMT5ForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxMT5Model: class FlaxMT5Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -843,6 +1077,9 @@ class FlaxMT5Model: ...@@ -843,6 +1077,9 @@ class FlaxMT5Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxPegasusForConditionalGeneration: class FlaxPegasusForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -852,6 +1089,9 @@ class FlaxPegasusForConditionalGeneration: ...@@ -852,6 +1089,9 @@ class FlaxPegasusForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxPegasusModel: class FlaxPegasusModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -861,6 +1101,9 @@ class FlaxPegasusModel: ...@@ -861,6 +1101,9 @@ class FlaxPegasusModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxPegasusPreTrainedModel: class FlaxPegasusPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -870,6 +1113,9 @@ class FlaxPegasusPreTrainedModel: ...@@ -870,6 +1113,9 @@ class FlaxPegasusPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxRobertaForMaskedLM: class FlaxRobertaForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -879,6 +1125,9 @@ class FlaxRobertaForMaskedLM: ...@@ -879,6 +1125,9 @@ class FlaxRobertaForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxRobertaForMultipleChoice: class FlaxRobertaForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -888,6 +1137,9 @@ class FlaxRobertaForMultipleChoice: ...@@ -888,6 +1137,9 @@ class FlaxRobertaForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxRobertaForQuestionAnswering: class FlaxRobertaForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -897,6 +1149,9 @@ class FlaxRobertaForQuestionAnswering: ...@@ -897,6 +1149,9 @@ class FlaxRobertaForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxRobertaForSequenceClassification: class FlaxRobertaForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -906,6 +1161,9 @@ class FlaxRobertaForSequenceClassification: ...@@ -906,6 +1161,9 @@ class FlaxRobertaForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxRobertaForTokenClassification: class FlaxRobertaForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -915,6 +1173,9 @@ class FlaxRobertaForTokenClassification: ...@@ -915,6 +1173,9 @@ class FlaxRobertaForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxRobertaModel: class FlaxRobertaModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -924,6 +1185,9 @@ class FlaxRobertaModel: ...@@ -924,6 +1185,9 @@ class FlaxRobertaModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxRobertaPreTrainedModel: class FlaxRobertaPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -933,6 +1197,9 @@ class FlaxRobertaPreTrainedModel: ...@@ -933,6 +1197,9 @@ class FlaxRobertaPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxT5ForConditionalGeneration: class FlaxT5ForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -942,6 +1209,9 @@ class FlaxT5ForConditionalGeneration: ...@@ -942,6 +1209,9 @@ class FlaxT5ForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxT5Model: class FlaxT5Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -951,6 +1221,9 @@ class FlaxT5Model: ...@@ -951,6 +1221,9 @@ class FlaxT5Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxT5PreTrainedModel: class FlaxT5PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -960,6 +1233,9 @@ class FlaxT5PreTrainedModel: ...@@ -960,6 +1233,9 @@ class FlaxT5PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxVisionEncoderDecoderModel: class FlaxVisionEncoderDecoderModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -969,6 +1245,9 @@ class FlaxVisionEncoderDecoderModel: ...@@ -969,6 +1245,9 @@ class FlaxVisionEncoderDecoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxViTForImageClassification: class FlaxViTForImageClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -983,6 +1262,9 @@ class FlaxViTModel: ...@@ -983,6 +1262,9 @@ class FlaxViTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxViTPreTrainedModel: class FlaxViTPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -992,6 +1274,9 @@ class FlaxViTPreTrainedModel: ...@@ -992,6 +1274,9 @@ class FlaxViTPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxWav2Vec2ForCTC: class FlaxWav2Vec2ForCTC:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1011,6 +1296,9 @@ class FlaxWav2Vec2Model: ...@@ -1011,6 +1296,9 @@ class FlaxWav2Vec2Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
class FlaxWav2Vec2PreTrainedModel: class FlaxWav2Vec2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1019,3 +1307,6 @@ class FlaxWav2Vec2PreTrainedModel: ...@@ -1019,3 +1307,6 @@ class FlaxWav2Vec2PreTrainedModel:
@classmethod @classmethod
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["flax"]) requires_backends(cls, ["flax"])
def __call__(self, *args, **kwargs):
requires_backends(self, ["flax"])
src/transformers/utils/dummy_pt_objects.py
View file @ 3e8d17e6
...@@ -223,6 +223,9 @@ class PreTrainedModel: ...@@ -223,6 +223,9 @@ class PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def apply_chunking_to_forward(*args, **kwargs): def apply_chunking_to_forward(*args, **kwargs):
requires_backends(apply_chunking_to_forward, ["torch"]) requires_backends(apply_chunking_to_forward, ["torch"])
...@@ -243,6 +246,9 @@ class AlbertForMaskedLM: ...@@ -243,6 +246,9 @@ class AlbertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AlbertForMultipleChoice: class AlbertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -252,6 +258,9 @@ class AlbertForMultipleChoice: ...@@ -252,6 +258,9 @@ class AlbertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AlbertForPreTraining: class AlbertForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -266,6 +275,9 @@ class AlbertForQuestionAnswering: ...@@ -266,6 +275,9 @@ class AlbertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AlbertForSequenceClassification: class AlbertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -275,6 +287,9 @@ class AlbertForSequenceClassification: ...@@ -275,6 +287,9 @@ class AlbertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AlbertForTokenClassification: class AlbertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -284,6 +299,9 @@ class AlbertForTokenClassification: ...@@ -284,6 +299,9 @@ class AlbertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AlbertModel: class AlbertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -293,6 +311,9 @@ class AlbertModel: ...@@ -293,6 +311,9 @@ class AlbertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AlbertPreTrainedModel: class AlbertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -302,6 +323,9 @@ class AlbertPreTrainedModel: ...@@ -302,6 +323,9 @@ class AlbertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_albert(*args, **kwargs): def load_tf_weights_in_albert(*args, **kwargs):
requires_backends(load_tf_weights_in_albert, ["torch"]) requires_backends(load_tf_weights_in_albert, ["torch"])
...@@ -372,6 +396,9 @@ class AutoModel: ...@@ -372,6 +396,9 @@ class AutoModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForAudioClassification: class AutoModelForAudioClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -381,6 +408,9 @@ class AutoModelForAudioClassification: ...@@ -381,6 +408,9 @@ class AutoModelForAudioClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForCausalLM: class AutoModelForCausalLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -390,6 +420,9 @@ class AutoModelForCausalLM: ...@@ -390,6 +420,9 @@ class AutoModelForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForCTC: class AutoModelForCTC:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -399,6 +432,9 @@ class AutoModelForCTC: ...@@ -399,6 +432,9 @@ class AutoModelForCTC:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForImageClassification: class AutoModelForImageClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -408,6 +444,9 @@ class AutoModelForImageClassification: ...@@ -408,6 +444,9 @@ class AutoModelForImageClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForImageSegmentation: class AutoModelForImageSegmentation:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -417,6 +456,9 @@ class AutoModelForImageSegmentation: ...@@ -417,6 +456,9 @@ class AutoModelForImageSegmentation:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForMaskedLM: class AutoModelForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -426,6 +468,9 @@ class AutoModelForMaskedLM: ...@@ -426,6 +468,9 @@ class AutoModelForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForMultipleChoice: class AutoModelForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -435,6 +480,9 @@ class AutoModelForMultipleChoice: ...@@ -435,6 +480,9 @@ class AutoModelForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForNextSentencePrediction: class AutoModelForNextSentencePrediction:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -444,6 +492,9 @@ class AutoModelForNextSentencePrediction: ...@@ -444,6 +492,9 @@ class AutoModelForNextSentencePrediction:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForObjectDetection: class AutoModelForObjectDetection:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -453,6 +504,9 @@ class AutoModelForObjectDetection: ...@@ -453,6 +504,9 @@ class AutoModelForObjectDetection:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForPreTraining: class AutoModelForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -462,6 +516,9 @@ class AutoModelForPreTraining: ...@@ -462,6 +516,9 @@ class AutoModelForPreTraining:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForQuestionAnswering: class AutoModelForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -471,6 +528,9 @@ class AutoModelForQuestionAnswering: ...@@ -471,6 +528,9 @@ class AutoModelForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForSeq2SeqLM: class AutoModelForSeq2SeqLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -480,6 +540,9 @@ class AutoModelForSeq2SeqLM: ...@@ -480,6 +540,9 @@ class AutoModelForSeq2SeqLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForSequenceClassification: class AutoModelForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -489,6 +552,9 @@ class AutoModelForSequenceClassification: ...@@ -489,6 +552,9 @@ class AutoModelForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForSpeechSeq2Seq: class AutoModelForSpeechSeq2Seq:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -498,6 +564,9 @@ class AutoModelForSpeechSeq2Seq: ...@@ -498,6 +564,9 @@ class AutoModelForSpeechSeq2Seq:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForTableQuestionAnswering: class AutoModelForTableQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -507,6 +576,9 @@ class AutoModelForTableQuestionAnswering: ...@@ -507,6 +576,9 @@ class AutoModelForTableQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForTokenClassification: class AutoModelForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -516,6 +588,9 @@ class AutoModelForTokenClassification: ...@@ -516,6 +588,9 @@ class AutoModelForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelForVision2Seq: class AutoModelForVision2Seq:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -525,6 +600,9 @@ class AutoModelForVision2Seq: ...@@ -525,6 +600,9 @@ class AutoModelForVision2Seq:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class AutoModelWithLMHead: class AutoModelWithLMHead:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -534,6 +612,9 @@ class AutoModelWithLMHead: ...@@ -534,6 +612,9 @@ class AutoModelWithLMHead:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
BART_PRETRAINED_MODEL_ARCHIVE_LIST = None BART_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -546,6 +627,9 @@ class BartForCausalLM: ...@@ -546,6 +627,9 @@ class BartForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BartForConditionalGeneration: class BartForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -555,6 +639,9 @@ class BartForConditionalGeneration: ...@@ -555,6 +639,9 @@ class BartForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BartForQuestionAnswering: class BartForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -564,6 +651,9 @@ class BartForQuestionAnswering: ...@@ -564,6 +651,9 @@ class BartForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BartForSequenceClassification: class BartForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -573,6 +663,9 @@ class BartForSequenceClassification: ...@@ -573,6 +663,9 @@ class BartForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BartModel: class BartModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -582,6 +675,9 @@ class BartModel: ...@@ -582,6 +675,9 @@ class BartModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BartPretrainedModel: class BartPretrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -591,6 +687,9 @@ class BartPretrainedModel: ...@@ -591,6 +687,9 @@ class BartPretrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class PretrainedBartModel: class PretrainedBartModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -600,6 +699,9 @@ class PretrainedBartModel: ...@@ -600,6 +699,9 @@ class PretrainedBartModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
BEIT_PRETRAINED_MODEL_ARCHIVE_LIST = None BEIT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -617,6 +719,9 @@ class BeitForMaskedImageModeling: ...@@ -617,6 +719,9 @@ class BeitForMaskedImageModeling:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BeitForSemanticSegmentation: class BeitForSemanticSegmentation:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -631,6 +736,9 @@ class BeitModel: ...@@ -631,6 +736,9 @@ class BeitModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BeitPreTrainedModel: class BeitPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -640,6 +748,9 @@ class BeitPreTrainedModel: ...@@ -640,6 +748,9 @@ class BeitPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -652,6 +763,9 @@ class BertForMaskedLM: ...@@ -652,6 +763,9 @@ class BertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BertForMultipleChoice: class BertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -661,6 +775,9 @@ class BertForMultipleChoice: ...@@ -661,6 +775,9 @@ class BertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BertForNextSentencePrediction: class BertForNextSentencePrediction:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -680,6 +797,9 @@ class BertForQuestionAnswering: ...@@ -680,6 +797,9 @@ class BertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BertForSequenceClassification: class BertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -689,6 +809,9 @@ class BertForSequenceClassification: ...@@ -689,6 +809,9 @@ class BertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BertForTokenClassification: class BertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -698,6 +821,9 @@ class BertForTokenClassification: ...@@ -698,6 +821,9 @@ class BertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BertLayer: class BertLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -712,6 +838,9 @@ class BertLMHeadModel: ...@@ -712,6 +838,9 @@ class BertLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BertModel: class BertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -721,6 +850,9 @@ class BertModel: ...@@ -721,6 +850,9 @@ class BertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BertPreTrainedModel: class BertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -730,6 +862,9 @@ class BertPreTrainedModel: ...@@ -730,6 +862,9 @@ class BertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_bert(*args, **kwargs): def load_tf_weights_in_bert(*args, **kwargs):
requires_backends(load_tf_weights_in_bert, ["torch"]) requires_backends(load_tf_weights_in_bert, ["torch"])
...@@ -753,6 +888,9 @@ class BertGenerationPreTrainedModel: ...@@ -753,6 +888,9 @@ class BertGenerationPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_bert_generation(*args, **kwargs): def load_tf_weights_in_bert_generation(*args, **kwargs):
requires_backends(load_tf_weights_in_bert_generation, ["torch"]) requires_backends(load_tf_weights_in_bert_generation, ["torch"])
...@@ -769,6 +907,9 @@ class BigBirdForCausalLM: ...@@ -769,6 +907,9 @@ class BigBirdForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdForMaskedLM: class BigBirdForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -778,6 +919,9 @@ class BigBirdForMaskedLM: ...@@ -778,6 +919,9 @@ class BigBirdForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdForMultipleChoice: class BigBirdForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -787,6 +931,9 @@ class BigBirdForMultipleChoice: ...@@ -787,6 +931,9 @@ class BigBirdForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdForPreTraining: class BigBirdForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -801,6 +948,9 @@ class BigBirdForQuestionAnswering: ...@@ -801,6 +948,9 @@ class BigBirdForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdForSequenceClassification: class BigBirdForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -810,6 +960,9 @@ class BigBirdForSequenceClassification: ...@@ -810,6 +960,9 @@ class BigBirdForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdForTokenClassification: class BigBirdForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -819,6 +972,9 @@ class BigBirdForTokenClassification: ...@@ -819,6 +972,9 @@ class BigBirdForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdLayer: class BigBirdLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -833,6 +989,9 @@ class BigBirdModel: ...@@ -833,6 +989,9 @@ class BigBirdModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdPreTrainedModel: class BigBirdPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -842,6 +1001,9 @@ class BigBirdPreTrainedModel: ...@@ -842,6 +1001,9 @@ class BigBirdPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_big_bird(*args, **kwargs): def load_tf_weights_in_big_bird(*args, **kwargs):
requires_backends(load_tf_weights_in_big_bird, ["torch"]) requires_backends(load_tf_weights_in_big_bird, ["torch"])
...@@ -858,6 +1020,9 @@ class BigBirdPegasusForCausalLM: ...@@ -858,6 +1020,9 @@ class BigBirdPegasusForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdPegasusForConditionalGeneration: class BigBirdPegasusForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -867,6 +1032,9 @@ class BigBirdPegasusForConditionalGeneration: ...@@ -867,6 +1032,9 @@ class BigBirdPegasusForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdPegasusForQuestionAnswering: class BigBirdPegasusForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -876,6 +1044,9 @@ class BigBirdPegasusForQuestionAnswering: ...@@ -876,6 +1044,9 @@ class BigBirdPegasusForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdPegasusForSequenceClassification: class BigBirdPegasusForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -885,6 +1056,9 @@ class BigBirdPegasusForSequenceClassification: ...@@ -885,6 +1056,9 @@ class BigBirdPegasusForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdPegasusModel: class BigBirdPegasusModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -894,6 +1068,9 @@ class BigBirdPegasusModel: ...@@ -894,6 +1068,9 @@ class BigBirdPegasusModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BigBirdPegasusPreTrainedModel: class BigBirdPegasusPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -903,6 +1080,9 @@ class BigBirdPegasusPreTrainedModel: ...@@ -903,6 +1080,9 @@ class BigBirdPegasusPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
BLENDERBOT_PRETRAINED_MODEL_ARCHIVE_LIST = None BLENDERBOT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -915,6 +1095,9 @@ class BlenderbotForCausalLM: ...@@ -915,6 +1095,9 @@ class BlenderbotForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BlenderbotForConditionalGeneration: class BlenderbotForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -924,6 +1107,9 @@ class BlenderbotForConditionalGeneration: ...@@ -924,6 +1107,9 @@ class BlenderbotForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BlenderbotModel: class BlenderbotModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -933,6 +1119,9 @@ class BlenderbotModel: ...@@ -933,6 +1119,9 @@ class BlenderbotModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BlenderbotPreTrainedModel: class BlenderbotPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -942,6 +1131,9 @@ class BlenderbotPreTrainedModel: ...@@ -942,6 +1131,9 @@ class BlenderbotPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
BLENDERBOT_SMALL_PRETRAINED_MODEL_ARCHIVE_LIST = None BLENDERBOT_SMALL_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -954,6 +1146,9 @@ class BlenderbotSmallForCausalLM: ...@@ -954,6 +1146,9 @@ class BlenderbotSmallForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BlenderbotSmallForConditionalGeneration: class BlenderbotSmallForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -963,6 +1158,9 @@ class BlenderbotSmallForConditionalGeneration: ...@@ -963,6 +1158,9 @@ class BlenderbotSmallForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BlenderbotSmallModel: class BlenderbotSmallModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -972,6 +1170,9 @@ class BlenderbotSmallModel: ...@@ -972,6 +1170,9 @@ class BlenderbotSmallModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class BlenderbotSmallPreTrainedModel: class BlenderbotSmallPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -981,6 +1182,9 @@ class BlenderbotSmallPreTrainedModel: ...@@ -981,6 +1182,9 @@ class BlenderbotSmallPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -993,6 +1197,9 @@ class CamembertForCausalLM: ...@@ -993,6 +1197,9 @@ class CamembertForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CamembertForMaskedLM: class CamembertForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1002,6 +1209,9 @@ class CamembertForMaskedLM: ...@@ -1002,6 +1209,9 @@ class CamembertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CamembertForMultipleChoice: class CamembertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1011,6 +1221,9 @@ class CamembertForMultipleChoice: ...@@ -1011,6 +1221,9 @@ class CamembertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CamembertForQuestionAnswering: class CamembertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1020,6 +1233,9 @@ class CamembertForQuestionAnswering: ...@@ -1020,6 +1233,9 @@ class CamembertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CamembertForSequenceClassification: class CamembertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1029,6 +1245,9 @@ class CamembertForSequenceClassification: ...@@ -1029,6 +1245,9 @@ class CamembertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CamembertForTokenClassification: class CamembertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1038,6 +1257,9 @@ class CamembertForTokenClassification: ...@@ -1038,6 +1257,9 @@ class CamembertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CamembertModel: class CamembertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1047,6 +1269,9 @@ class CamembertModel: ...@@ -1047,6 +1269,9 @@ class CamembertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
CANINE_PRETRAINED_MODEL_ARCHIVE_LIST = None CANINE_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1059,6 +1284,9 @@ class CanineForMultipleChoice: ...@@ -1059,6 +1284,9 @@ class CanineForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CanineForQuestionAnswering: class CanineForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1068,6 +1296,9 @@ class CanineForQuestionAnswering: ...@@ -1068,6 +1296,9 @@ class CanineForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CanineForSequenceClassification: class CanineForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1077,6 +1308,9 @@ class CanineForSequenceClassification: ...@@ -1077,6 +1308,9 @@ class CanineForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CanineForTokenClassification: class CanineForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1086,6 +1320,9 @@ class CanineForTokenClassification: ...@@ -1086,6 +1320,9 @@ class CanineForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CanineLayer: class CanineLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1100,6 +1337,9 @@ class CanineModel: ...@@ -1100,6 +1337,9 @@ class CanineModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CaninePreTrainedModel: class CaninePreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1109,6 +1349,9 @@ class CaninePreTrainedModel: ...@@ -1109,6 +1349,9 @@ class CaninePreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_canine(*args, **kwargs): def load_tf_weights_in_canine(*args, **kwargs):
requires_backends(load_tf_weights_in_canine, ["torch"]) requires_backends(load_tf_weights_in_canine, ["torch"])
...@@ -1125,6 +1368,9 @@ class CLIPModel: ...@@ -1125,6 +1368,9 @@ class CLIPModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CLIPPreTrainedModel: class CLIPPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1134,6 +1380,9 @@ class CLIPPreTrainedModel: ...@@ -1134,6 +1380,9 @@ class CLIPPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CLIPTextModel: class CLIPTextModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1143,6 +1392,9 @@ class CLIPTextModel: ...@@ -1143,6 +1392,9 @@ class CLIPTextModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CLIPVisionModel: class CLIPVisionModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1152,6 +1404,9 @@ class CLIPVisionModel: ...@@ -1152,6 +1404,9 @@ class CLIPVisionModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
CONVBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None CONVBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1164,6 +1419,9 @@ class ConvBertForMaskedLM: ...@@ -1164,6 +1419,9 @@ class ConvBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ConvBertForMultipleChoice: class ConvBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1173,6 +1431,9 @@ class ConvBertForMultipleChoice: ...@@ -1173,6 +1431,9 @@ class ConvBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ConvBertForQuestionAnswering: class ConvBertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1182,6 +1443,9 @@ class ConvBertForQuestionAnswering: ...@@ -1182,6 +1443,9 @@ class ConvBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ConvBertForSequenceClassification: class ConvBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1191,6 +1455,9 @@ class ConvBertForSequenceClassification: ...@@ -1191,6 +1455,9 @@ class ConvBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ConvBertForTokenClassification: class ConvBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1200,6 +1467,9 @@ class ConvBertForTokenClassification: ...@@ -1200,6 +1467,9 @@ class ConvBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ConvBertLayer: class ConvBertLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1214,6 +1484,9 @@ class ConvBertModel: ...@@ -1214,6 +1484,9 @@ class ConvBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ConvBertPreTrainedModel: class ConvBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1223,6 +1496,9 @@ class ConvBertPreTrainedModel: ...@@ -1223,6 +1496,9 @@ class ConvBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_convbert(*args, **kwargs): def load_tf_weights_in_convbert(*args, **kwargs):
requires_backends(load_tf_weights_in_convbert, ["torch"]) requires_backends(load_tf_weights_in_convbert, ["torch"])
...@@ -1239,6 +1515,9 @@ class CTRLForSequenceClassification: ...@@ -1239,6 +1515,9 @@ class CTRLForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CTRLLMHeadModel: class CTRLLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1248,6 +1527,9 @@ class CTRLLMHeadModel: ...@@ -1248,6 +1527,9 @@ class CTRLLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CTRLModel: class CTRLModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1257,6 +1539,9 @@ class CTRLModel: ...@@ -1257,6 +1539,9 @@ class CTRLModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class CTRLPreTrainedModel: class CTRLPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1266,6 +1551,9 @@ class CTRLPreTrainedModel: ...@@ -1266,6 +1551,9 @@ class CTRLPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
DEBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None DEBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1278,6 +1566,9 @@ class DebertaForMaskedLM: ...@@ -1278,6 +1566,9 @@ class DebertaForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaForQuestionAnswering: class DebertaForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1287,6 +1578,9 @@ class DebertaForQuestionAnswering: ...@@ -1287,6 +1578,9 @@ class DebertaForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaForSequenceClassification: class DebertaForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1296,6 +1590,9 @@ class DebertaForSequenceClassification: ...@@ -1296,6 +1590,9 @@ class DebertaForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaForTokenClassification: class DebertaForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1305,6 +1602,9 @@ class DebertaForTokenClassification: ...@@ -1305,6 +1602,9 @@ class DebertaForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaModel: class DebertaModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1314,6 +1614,9 @@ class DebertaModel: ...@@ -1314,6 +1614,9 @@ class DebertaModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaPreTrainedModel: class DebertaPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1323,6 +1626,9 @@ class DebertaPreTrainedModel: ...@@ -1323,6 +1626,9 @@ class DebertaPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
DEBERTA_V2_PRETRAINED_MODEL_ARCHIVE_LIST = None DEBERTA_V2_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1335,6 +1641,9 @@ class DebertaV2ForMaskedLM: ...@@ -1335,6 +1641,9 @@ class DebertaV2ForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaV2ForQuestionAnswering: class DebertaV2ForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1344,6 +1653,9 @@ class DebertaV2ForQuestionAnswering: ...@@ -1344,6 +1653,9 @@ class DebertaV2ForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaV2ForSequenceClassification: class DebertaV2ForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1353,6 +1665,9 @@ class DebertaV2ForSequenceClassification: ...@@ -1353,6 +1665,9 @@ class DebertaV2ForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaV2ForTokenClassification: class DebertaV2ForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1362,6 +1677,9 @@ class DebertaV2ForTokenClassification: ...@@ -1362,6 +1677,9 @@ class DebertaV2ForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaV2Model: class DebertaV2Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1371,6 +1689,9 @@ class DebertaV2Model: ...@@ -1371,6 +1689,9 @@ class DebertaV2Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DebertaV2PreTrainedModel: class DebertaV2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1380,6 +1701,9 @@ class DebertaV2PreTrainedModel: ...@@ -1380,6 +1701,9 @@ class DebertaV2PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
DEIT_PRETRAINED_MODEL_ARCHIVE_LIST = None DEIT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1402,6 +1726,9 @@ class DeiTModel: ...@@ -1402,6 +1726,9 @@ class DeiTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DeiTPreTrainedModel: class DeiTPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1411,6 +1738,9 @@ class DeiTPreTrainedModel: ...@@ -1411,6 +1738,9 @@ class DeiTPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1423,6 +1753,9 @@ class DistilBertForMaskedLM: ...@@ -1423,6 +1753,9 @@ class DistilBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DistilBertForMultipleChoice: class DistilBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1432,6 +1765,9 @@ class DistilBertForMultipleChoice: ...@@ -1432,6 +1765,9 @@ class DistilBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DistilBertForQuestionAnswering: class DistilBertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1441,6 +1777,9 @@ class DistilBertForQuestionAnswering: ...@@ -1441,6 +1777,9 @@ class DistilBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DistilBertForSequenceClassification: class DistilBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1450,6 +1789,9 @@ class DistilBertForSequenceClassification: ...@@ -1450,6 +1789,9 @@ class DistilBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DistilBertForTokenClassification: class DistilBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1459,6 +1801,9 @@ class DistilBertForTokenClassification: ...@@ -1459,6 +1801,9 @@ class DistilBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DistilBertModel: class DistilBertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1468,6 +1813,9 @@ class DistilBertModel: ...@@ -1468,6 +1813,9 @@ class DistilBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DistilBertPreTrainedModel: class DistilBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1477,6 +1825,9 @@ class DistilBertPreTrainedModel: ...@@ -1477,6 +1825,9 @@ class DistilBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1505,6 +1856,9 @@ class DPRPreTrainedModel: ...@@ -1505,6 +1856,9 @@ class DPRPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class DPRPretrainedQuestionEncoder: class DPRPretrainedQuestionEncoder:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1537,6 +1891,9 @@ class ElectraForMaskedLM: ...@@ -1537,6 +1891,9 @@ class ElectraForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ElectraForMultipleChoice: class ElectraForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1546,6 +1903,9 @@ class ElectraForMultipleChoice: ...@@ -1546,6 +1903,9 @@ class ElectraForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ElectraForPreTraining: class ElectraForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1560,6 +1920,9 @@ class ElectraForQuestionAnswering: ...@@ -1560,6 +1920,9 @@ class ElectraForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ElectraForSequenceClassification: class ElectraForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1569,6 +1932,9 @@ class ElectraForSequenceClassification: ...@@ -1569,6 +1932,9 @@ class ElectraForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ElectraForTokenClassification: class ElectraForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1578,6 +1944,9 @@ class ElectraForTokenClassification: ...@@ -1578,6 +1944,9 @@ class ElectraForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ElectraModel: class ElectraModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1587,6 +1956,9 @@ class ElectraModel: ...@@ -1587,6 +1956,9 @@ class ElectraModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ElectraPreTrainedModel: class ElectraPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1596,6 +1968,9 @@ class ElectraPreTrainedModel: ...@@ -1596,6 +1968,9 @@ class ElectraPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_electra(*args, **kwargs): def load_tf_weights_in_electra(*args, **kwargs):
requires_backends(load_tf_weights_in_electra, ["torch"]) requires_backends(load_tf_weights_in_electra, ["torch"])
...@@ -1609,6 +1984,9 @@ class EncoderDecoderModel: ...@@ -1609,6 +1984,9 @@ class EncoderDecoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1621,6 +1999,9 @@ class FlaubertForMultipleChoice: ...@@ -1621,6 +1999,9 @@ class FlaubertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FlaubertForQuestionAnswering: class FlaubertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1630,6 +2011,9 @@ class FlaubertForQuestionAnswering: ...@@ -1630,6 +2011,9 @@ class FlaubertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FlaubertForQuestionAnsweringSimple: class FlaubertForQuestionAnsweringSimple:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1639,6 +2023,9 @@ class FlaubertForQuestionAnsweringSimple: ...@@ -1639,6 +2023,9 @@ class FlaubertForQuestionAnsweringSimple:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FlaubertForSequenceClassification: class FlaubertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1648,6 +2035,9 @@ class FlaubertForSequenceClassification: ...@@ -1648,6 +2035,9 @@ class FlaubertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FlaubertForTokenClassification: class FlaubertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1657,6 +2047,9 @@ class FlaubertForTokenClassification: ...@@ -1657,6 +2047,9 @@ class FlaubertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FlaubertModel: class FlaubertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1666,6 +2059,9 @@ class FlaubertModel: ...@@ -1666,6 +2059,9 @@ class FlaubertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FlaubertWithLMHeadModel: class FlaubertWithLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1675,6 +2071,9 @@ class FlaubertWithLMHeadModel: ...@@ -1675,6 +2071,9 @@ class FlaubertWithLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
FNET_PRETRAINED_MODEL_ARCHIVE_LIST = None FNET_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1687,6 +2086,9 @@ class FNetForMaskedLM: ...@@ -1687,6 +2086,9 @@ class FNetForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FNetForMultipleChoice: class FNetForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1696,6 +2098,9 @@ class FNetForMultipleChoice: ...@@ -1696,6 +2098,9 @@ class FNetForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FNetForNextSentencePrediction: class FNetForNextSentencePrediction:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1715,6 +2120,9 @@ class FNetForQuestionAnswering: ...@@ -1715,6 +2120,9 @@ class FNetForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FNetForSequenceClassification: class FNetForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1724,6 +2132,9 @@ class FNetForSequenceClassification: ...@@ -1724,6 +2132,9 @@ class FNetForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FNetForTokenClassification: class FNetForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1733,6 +2144,9 @@ class FNetForTokenClassification: ...@@ -1733,6 +2144,9 @@ class FNetForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FNetLayer: class FNetLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1747,6 +2161,9 @@ class FNetModel: ...@@ -1747,6 +2161,9 @@ class FNetModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FNetPreTrainedModel: class FNetPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1756,6 +2173,9 @@ class FNetPreTrainedModel: ...@@ -1756,6 +2173,9 @@ class FNetPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FSMTForConditionalGeneration: class FSMTForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1765,6 +2185,9 @@ class FSMTForConditionalGeneration: ...@@ -1765,6 +2185,9 @@ class FSMTForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FSMTModel: class FSMTModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1774,6 +2197,9 @@ class FSMTModel: ...@@ -1774,6 +2197,9 @@ class FSMTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class PretrainedFSMTModel: class PretrainedFSMTModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1783,6 +2209,9 @@ class PretrainedFSMTModel: ...@@ -1783,6 +2209,9 @@ class PretrainedFSMTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
FUNNEL_PRETRAINED_MODEL_ARCHIVE_LIST = None FUNNEL_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1795,6 +2224,9 @@ class FunnelBaseModel: ...@@ -1795,6 +2224,9 @@ class FunnelBaseModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FunnelForMaskedLM: class FunnelForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1804,6 +2236,9 @@ class FunnelForMaskedLM: ...@@ -1804,6 +2236,9 @@ class FunnelForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FunnelForMultipleChoice: class FunnelForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1813,6 +2248,9 @@ class FunnelForMultipleChoice: ...@@ -1813,6 +2248,9 @@ class FunnelForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FunnelForPreTraining: class FunnelForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1827,6 +2265,9 @@ class FunnelForQuestionAnswering: ...@@ -1827,6 +2265,9 @@ class FunnelForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FunnelForSequenceClassification: class FunnelForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1836,6 +2277,9 @@ class FunnelForSequenceClassification: ...@@ -1836,6 +2277,9 @@ class FunnelForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FunnelForTokenClassification: class FunnelForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1845,6 +2289,9 @@ class FunnelForTokenClassification: ...@@ -1845,6 +2289,9 @@ class FunnelForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FunnelModel: class FunnelModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1854,6 +2301,9 @@ class FunnelModel: ...@@ -1854,6 +2301,9 @@ class FunnelModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class FunnelPreTrainedModel: class FunnelPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1863,6 +2313,9 @@ class FunnelPreTrainedModel: ...@@ -1863,6 +2313,9 @@ class FunnelPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_funnel(*args, **kwargs): def load_tf_weights_in_funnel(*args, **kwargs):
requires_backends(load_tf_weights_in_funnel, ["torch"]) requires_backends(load_tf_weights_in_funnel, ["torch"])
...@@ -1879,6 +2332,9 @@ class GPT2DoubleHeadsModel: ...@@ -1879,6 +2332,9 @@ class GPT2DoubleHeadsModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPT2ForSequenceClassification: class GPT2ForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1888,6 +2344,9 @@ class GPT2ForSequenceClassification: ...@@ -1888,6 +2344,9 @@ class GPT2ForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPT2ForTokenClassification: class GPT2ForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1897,6 +2356,9 @@ class GPT2ForTokenClassification: ...@@ -1897,6 +2356,9 @@ class GPT2ForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPT2LMHeadModel: class GPT2LMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1906,6 +2368,9 @@ class GPT2LMHeadModel: ...@@ -1906,6 +2368,9 @@ class GPT2LMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPT2Model: class GPT2Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1915,6 +2380,9 @@ class GPT2Model: ...@@ -1915,6 +2380,9 @@ class GPT2Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPT2PreTrainedModel: class GPT2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1924,6 +2392,9 @@ class GPT2PreTrainedModel: ...@@ -1924,6 +2392,9 @@ class GPT2PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_gpt2(*args, **kwargs): def load_tf_weights_in_gpt2(*args, **kwargs):
requires_backends(load_tf_weights_in_gpt2, ["torch"]) requires_backends(load_tf_weights_in_gpt2, ["torch"])
...@@ -1940,6 +2411,9 @@ class GPTNeoForCausalLM: ...@@ -1940,6 +2411,9 @@ class GPTNeoForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPTNeoForSequenceClassification: class GPTNeoForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1949,6 +2423,9 @@ class GPTNeoForSequenceClassification: ...@@ -1949,6 +2423,9 @@ class GPTNeoForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPTNeoModel: class GPTNeoModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1958,6 +2435,9 @@ class GPTNeoModel: ...@@ -1958,6 +2435,9 @@ class GPTNeoModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPTNeoPreTrainedModel: class GPTNeoPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1967,6 +2447,9 @@ class GPTNeoPreTrainedModel: ...@@ -1967,6 +2447,9 @@ class GPTNeoPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_gpt_neo(*args, **kwargs): def load_tf_weights_in_gpt_neo(*args, **kwargs):
requires_backends(load_tf_weights_in_gpt_neo, ["torch"]) requires_backends(load_tf_weights_in_gpt_neo, ["torch"])
...@@ -1983,6 +2466,9 @@ class GPTJForCausalLM: ...@@ -1983,6 +2466,9 @@ class GPTJForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPTJForSequenceClassification: class GPTJForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1992,6 +2478,9 @@ class GPTJForSequenceClassification: ...@@ -1992,6 +2478,9 @@ class GPTJForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPTJModel: class GPTJModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2001,6 +2490,9 @@ class GPTJModel: ...@@ -2001,6 +2490,9 @@ class GPTJModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class GPTJPreTrainedModel: class GPTJPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2010,6 +2502,9 @@ class GPTJPreTrainedModel: ...@@ -2010,6 +2502,9 @@ class GPTJPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2027,6 +2522,9 @@ class HubertForSequenceClassification: ...@@ -2027,6 +2522,9 @@ class HubertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class HubertModel: class HubertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2036,6 +2534,9 @@ class HubertModel: ...@@ -2036,6 +2534,9 @@ class HubertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class HubertPreTrainedModel: class HubertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2045,6 +2546,9 @@ class HubertPreTrainedModel: ...@@ -2045,6 +2546,9 @@ class HubertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
IBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None IBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2057,6 +2561,9 @@ class IBertForMaskedLM: ...@@ -2057,6 +2561,9 @@ class IBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class IBertForMultipleChoice: class IBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2066,6 +2573,9 @@ class IBertForMultipleChoice: ...@@ -2066,6 +2573,9 @@ class IBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class IBertForQuestionAnswering: class IBertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2075,6 +2585,9 @@ class IBertForQuestionAnswering: ...@@ -2075,6 +2585,9 @@ class IBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class IBertForSequenceClassification: class IBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2084,6 +2597,9 @@ class IBertForSequenceClassification: ...@@ -2084,6 +2597,9 @@ class IBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class IBertForTokenClassification: class IBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2093,6 +2609,9 @@ class IBertForTokenClassification: ...@@ -2093,6 +2609,9 @@ class IBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class IBertModel: class IBertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2102,6 +2621,9 @@ class IBertModel: ...@@ -2102,6 +2621,9 @@ class IBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class IBertPreTrainedModel: class IBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2111,6 +2633,9 @@ class IBertPreTrainedModel: ...@@ -2111,6 +2633,9 @@ class IBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST = None LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2123,6 +2648,9 @@ class LayoutLMForMaskedLM: ...@@ -2123,6 +2648,9 @@ class LayoutLMForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LayoutLMForSequenceClassification: class LayoutLMForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2132,6 +2660,9 @@ class LayoutLMForSequenceClassification: ...@@ -2132,6 +2660,9 @@ class LayoutLMForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LayoutLMForTokenClassification: class LayoutLMForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2141,6 +2672,9 @@ class LayoutLMForTokenClassification: ...@@ -2141,6 +2672,9 @@ class LayoutLMForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LayoutLMModel: class LayoutLMModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2150,6 +2684,9 @@ class LayoutLMModel: ...@@ -2150,6 +2684,9 @@ class LayoutLMModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LayoutLMPreTrainedModel: class LayoutLMPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2159,6 +2696,9 @@ class LayoutLMPreTrainedModel: ...@@ -2159,6 +2696,9 @@ class LayoutLMPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
LAYOUTLMV2_PRETRAINED_MODEL_ARCHIVE_LIST = None LAYOUTLMV2_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2171,6 +2711,9 @@ class LayoutLMv2ForQuestionAnswering: ...@@ -2171,6 +2711,9 @@ class LayoutLMv2ForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LayoutLMv2ForSequenceClassification: class LayoutLMv2ForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2180,6 +2723,9 @@ class LayoutLMv2ForSequenceClassification: ...@@ -2180,6 +2723,9 @@ class LayoutLMv2ForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LayoutLMv2ForTokenClassification: class LayoutLMv2ForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2189,6 +2735,9 @@ class LayoutLMv2ForTokenClassification: ...@@ -2189,6 +2735,9 @@ class LayoutLMv2ForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LayoutLMv2Model: class LayoutLMv2Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2198,6 +2747,9 @@ class LayoutLMv2Model: ...@@ -2198,6 +2747,9 @@ class LayoutLMv2Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LayoutLMv2PreTrainedModel: class LayoutLMv2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2207,6 +2759,9 @@ class LayoutLMv2PreTrainedModel: ...@@ -2207,6 +2759,9 @@ class LayoutLMv2PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
LED_PRETRAINED_MODEL_ARCHIVE_LIST = None LED_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2219,6 +2774,9 @@ class LEDForConditionalGeneration: ...@@ -2219,6 +2774,9 @@ class LEDForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LEDForQuestionAnswering: class LEDForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2228,6 +2786,9 @@ class LEDForQuestionAnswering: ...@@ -2228,6 +2786,9 @@ class LEDForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LEDForSequenceClassification: class LEDForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2237,6 +2798,9 @@ class LEDForSequenceClassification: ...@@ -2237,6 +2798,9 @@ class LEDForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LEDModel: class LEDModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2246,6 +2810,9 @@ class LEDModel: ...@@ -2246,6 +2810,9 @@ class LEDModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LEDPreTrainedModel: class LEDPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2255,6 +2822,9 @@ class LEDPreTrainedModel: ...@@ -2255,6 +2822,9 @@ class LEDPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2267,6 +2837,9 @@ class LongformerForMaskedLM: ...@@ -2267,6 +2837,9 @@ class LongformerForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LongformerForMultipleChoice: class LongformerForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2276,6 +2849,9 @@ class LongformerForMultipleChoice: ...@@ -2276,6 +2849,9 @@ class LongformerForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LongformerForQuestionAnswering: class LongformerForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2285,6 +2861,9 @@ class LongformerForQuestionAnswering: ...@@ -2285,6 +2861,9 @@ class LongformerForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LongformerForSequenceClassification: class LongformerForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2294,6 +2873,9 @@ class LongformerForSequenceClassification: ...@@ -2294,6 +2873,9 @@ class LongformerForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LongformerForTokenClassification: class LongformerForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2303,6 +2885,9 @@ class LongformerForTokenClassification: ...@@ -2303,6 +2885,9 @@ class LongformerForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LongformerModel: class LongformerModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2312,6 +2897,9 @@ class LongformerModel: ...@@ -2312,6 +2897,9 @@ class LongformerModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LongformerPreTrainedModel: class LongformerPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2321,6 +2909,9 @@ class LongformerPreTrainedModel: ...@@ -2321,6 +2909,9 @@ class LongformerPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LongformerSelfAttention: class LongformerSelfAttention:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2353,6 +2944,9 @@ class LukeModel: ...@@ -2353,6 +2944,9 @@ class LukeModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LukePreTrainedModel: class LukePreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2362,6 +2956,9 @@ class LukePreTrainedModel: ...@@ -2362,6 +2956,9 @@ class LukePreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LxmertEncoder: class LxmertEncoder:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2381,6 +2978,9 @@ class LxmertForQuestionAnswering: ...@@ -2381,6 +2978,9 @@ class LxmertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LxmertModel: class LxmertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2390,6 +2990,9 @@ class LxmertModel: ...@@ -2390,6 +2990,9 @@ class LxmertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LxmertPreTrainedModel: class LxmertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2399,6 +3002,9 @@ class LxmertPreTrainedModel: ...@@ -2399,6 +3002,9 @@ class LxmertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class LxmertVisualFeatureEncoder: class LxmertVisualFeatureEncoder:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2421,6 +3027,9 @@ class M2M100ForConditionalGeneration: ...@@ -2421,6 +3027,9 @@ class M2M100ForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class M2M100Model: class M2M100Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2430,6 +3039,9 @@ class M2M100Model: ...@@ -2430,6 +3039,9 @@ class M2M100Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class M2M100PreTrainedModel: class M2M100PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2439,6 +3051,9 @@ class M2M100PreTrainedModel: ...@@ -2439,6 +3051,9 @@ class M2M100PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MarianForCausalLM: class MarianForCausalLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2448,6 +3063,9 @@ class MarianForCausalLM: ...@@ -2448,6 +3063,9 @@ class MarianForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MarianModel: class MarianModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2457,6 +3075,9 @@ class MarianModel: ...@@ -2457,6 +3075,9 @@ class MarianModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MarianMTModel: class MarianMTModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2466,6 +3087,9 @@ class MarianMTModel: ...@@ -2466,6 +3087,9 @@ class MarianMTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MBartForCausalLM: class MBartForCausalLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2475,6 +3099,9 @@ class MBartForCausalLM: ...@@ -2475,6 +3099,9 @@ class MBartForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MBartForConditionalGeneration: class MBartForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2484,6 +3111,9 @@ class MBartForConditionalGeneration: ...@@ -2484,6 +3111,9 @@ class MBartForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MBartForQuestionAnswering: class MBartForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2493,6 +3123,9 @@ class MBartForQuestionAnswering: ...@@ -2493,6 +3123,9 @@ class MBartForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MBartForSequenceClassification: class MBartForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2502,6 +3135,9 @@ class MBartForSequenceClassification: ...@@ -2502,6 +3135,9 @@ class MBartForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MBartModel: class MBartModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2511,6 +3147,9 @@ class MBartModel: ...@@ -2511,6 +3147,9 @@ class MBartModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MBartPreTrainedModel: class MBartPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2520,6 +3159,9 @@ class MBartPreTrainedModel: ...@@ -2520,6 +3159,9 @@ class MBartPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2532,6 +3174,9 @@ class MegatronBertForCausalLM: ...@@ -2532,6 +3174,9 @@ class MegatronBertForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MegatronBertForMaskedLM: class MegatronBertForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2541,6 +3186,9 @@ class MegatronBertForMaskedLM: ...@@ -2541,6 +3186,9 @@ class MegatronBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MegatronBertForMultipleChoice: class MegatronBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2550,6 +3198,9 @@ class MegatronBertForMultipleChoice: ...@@ -2550,6 +3198,9 @@ class MegatronBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MegatronBertForNextSentencePrediction: class MegatronBertForNextSentencePrediction:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2569,6 +3220,9 @@ class MegatronBertForQuestionAnswering: ...@@ -2569,6 +3220,9 @@ class MegatronBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MegatronBertForSequenceClassification: class MegatronBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2578,6 +3232,9 @@ class MegatronBertForSequenceClassification: ...@@ -2578,6 +3232,9 @@ class MegatronBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MegatronBertForTokenClassification: class MegatronBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2587,6 +3244,9 @@ class MegatronBertForTokenClassification: ...@@ -2587,6 +3244,9 @@ class MegatronBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MegatronBertModel: class MegatronBertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2596,6 +3256,9 @@ class MegatronBertModel: ...@@ -2596,6 +3256,9 @@ class MegatronBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MegatronBertPreTrainedModel: class MegatronBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2605,6 +3268,9 @@ class MegatronBertPreTrainedModel: ...@@ -2605,6 +3268,9 @@ class MegatronBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MMBTForClassification: class MMBTForClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2619,6 +3285,9 @@ class MMBTModel: ...@@ -2619,6 +3285,9 @@ class MMBTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ModalEmbeddings: class ModalEmbeddings:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2636,6 +3305,9 @@ class MobileBertForMaskedLM: ...@@ -2636,6 +3305,9 @@ class MobileBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MobileBertForMultipleChoice: class MobileBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2645,6 +3317,9 @@ class MobileBertForMultipleChoice: ...@@ -2645,6 +3317,9 @@ class MobileBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MobileBertForNextSentencePrediction: class MobileBertForNextSentencePrediction:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2664,6 +3339,9 @@ class MobileBertForQuestionAnswering: ...@@ -2664,6 +3339,9 @@ class MobileBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MobileBertForSequenceClassification: class MobileBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2673,6 +3351,9 @@ class MobileBertForSequenceClassification: ...@@ -2673,6 +3351,9 @@ class MobileBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MobileBertForTokenClassification: class MobileBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2682,6 +3363,9 @@ class MobileBertForTokenClassification: ...@@ -2682,6 +3363,9 @@ class MobileBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MobileBertLayer: class MobileBertLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2696,6 +3380,9 @@ class MobileBertModel: ...@@ -2696,6 +3380,9 @@ class MobileBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MobileBertPreTrainedModel: class MobileBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2705,6 +3392,9 @@ class MobileBertPreTrainedModel: ...@@ -2705,6 +3392,9 @@ class MobileBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_mobilebert(*args, **kwargs): def load_tf_weights_in_mobilebert(*args, **kwargs):
requires_backends(load_tf_weights_in_mobilebert, ["torch"]) requires_backends(load_tf_weights_in_mobilebert, ["torch"])
...@@ -2721,6 +3411,9 @@ class MPNetForMaskedLM: ...@@ -2721,6 +3411,9 @@ class MPNetForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MPNetForMultipleChoice: class MPNetForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2730,6 +3423,9 @@ class MPNetForMultipleChoice: ...@@ -2730,6 +3423,9 @@ class MPNetForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MPNetForQuestionAnswering: class MPNetForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2739,6 +3435,9 @@ class MPNetForQuestionAnswering: ...@@ -2739,6 +3435,9 @@ class MPNetForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MPNetForSequenceClassification: class MPNetForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2748,6 +3447,9 @@ class MPNetForSequenceClassification: ...@@ -2748,6 +3447,9 @@ class MPNetForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MPNetForTokenClassification: class MPNetForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2757,6 +3459,9 @@ class MPNetForTokenClassification: ...@@ -2757,6 +3459,9 @@ class MPNetForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MPNetLayer: class MPNetLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2771,6 +3476,9 @@ class MPNetModel: ...@@ -2771,6 +3476,9 @@ class MPNetModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MPNetPreTrainedModel: class MPNetPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2780,6 +3488,9 @@ class MPNetPreTrainedModel: ...@@ -2780,6 +3488,9 @@ class MPNetPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MT5EncoderModel: class MT5EncoderModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2789,6 +3500,9 @@ class MT5EncoderModel: ...@@ -2789,6 +3500,9 @@ class MT5EncoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MT5ForConditionalGeneration: class MT5ForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2798,6 +3512,9 @@ class MT5ForConditionalGeneration: ...@@ -2798,6 +3512,9 @@ class MT5ForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class MT5Model: class MT5Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2807,6 +3524,9 @@ class MT5Model: ...@@ -2807,6 +3524,9 @@ class MT5Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST = None OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2819,6 +3539,9 @@ class OpenAIGPTDoubleHeadsModel: ...@@ -2819,6 +3539,9 @@ class OpenAIGPTDoubleHeadsModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class OpenAIGPTForSequenceClassification: class OpenAIGPTForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2828,6 +3551,9 @@ class OpenAIGPTForSequenceClassification: ...@@ -2828,6 +3551,9 @@ class OpenAIGPTForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class OpenAIGPTLMHeadModel: class OpenAIGPTLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2837,6 +3563,9 @@ class OpenAIGPTLMHeadModel: ...@@ -2837,6 +3563,9 @@ class OpenAIGPTLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class OpenAIGPTModel: class OpenAIGPTModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2846,6 +3575,9 @@ class OpenAIGPTModel: ...@@ -2846,6 +3575,9 @@ class OpenAIGPTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class OpenAIGPTPreTrainedModel: class OpenAIGPTPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2855,6 +3587,9 @@ class OpenAIGPTPreTrainedModel: ...@@ -2855,6 +3587,9 @@ class OpenAIGPTPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_openai_gpt(*args, **kwargs): def load_tf_weights_in_openai_gpt(*args, **kwargs):
requires_backends(load_tf_weights_in_openai_gpt, ["torch"]) requires_backends(load_tf_weights_in_openai_gpt, ["torch"])
...@@ -2868,6 +3603,9 @@ class PegasusForCausalLM: ...@@ -2868,6 +3603,9 @@ class PegasusForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class PegasusForConditionalGeneration: class PegasusForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2877,6 +3615,9 @@ class PegasusForConditionalGeneration: ...@@ -2877,6 +3615,9 @@ class PegasusForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class PegasusModel: class PegasusModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2886,6 +3627,9 @@ class PegasusModel: ...@@ -2886,6 +3627,9 @@ class PegasusModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class PegasusPreTrainedModel: class PegasusPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2895,6 +3639,9 @@ class PegasusPreTrainedModel: ...@@ -2895,6 +3639,9 @@ class PegasusPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LIST = None PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2917,6 +3664,9 @@ class ProphetNetForCausalLM: ...@@ -2917,6 +3664,9 @@ class ProphetNetForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ProphetNetForConditionalGeneration: class ProphetNetForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2926,6 +3676,9 @@ class ProphetNetForConditionalGeneration: ...@@ -2926,6 +3676,9 @@ class ProphetNetForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ProphetNetModel: class ProphetNetModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2935,6 +3688,9 @@ class ProphetNetModel: ...@@ -2935,6 +3688,9 @@ class ProphetNetModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ProphetNetPreTrainedModel: class ProphetNetPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2944,6 +3700,9 @@ class ProphetNetPreTrainedModel: ...@@ -2944,6 +3700,9 @@ class ProphetNetPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RagModel: class RagModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2953,6 +3712,9 @@ class RagModel: ...@@ -2953,6 +3712,9 @@ class RagModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RagPreTrainedModel: class RagPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2962,6 +3724,9 @@ class RagPreTrainedModel: ...@@ -2962,6 +3724,9 @@ class RagPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RagSequenceForGeneration: class RagSequenceForGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2989,6 +3754,9 @@ class ReformerForMaskedLM: ...@@ -2989,6 +3754,9 @@ class ReformerForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ReformerForQuestionAnswering: class ReformerForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2998,6 +3766,9 @@ class ReformerForQuestionAnswering: ...@@ -2998,6 +3766,9 @@ class ReformerForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ReformerForSequenceClassification: class ReformerForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3007,6 +3778,9 @@ class ReformerForSequenceClassification: ...@@ -3007,6 +3778,9 @@ class ReformerForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ReformerLayer: class ReformerLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3021,6 +3795,9 @@ class ReformerModel: ...@@ -3021,6 +3795,9 @@ class ReformerModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ReformerModelWithLMHead: class ReformerModelWithLMHead:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3030,6 +3807,9 @@ class ReformerModelWithLMHead: ...@@ -3030,6 +3807,9 @@ class ReformerModelWithLMHead:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ReformerPreTrainedModel: class ReformerPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3039,6 +3819,9 @@ class ReformerPreTrainedModel: ...@@ -3039,6 +3819,9 @@ class ReformerPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
REMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None REMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3051,6 +3834,9 @@ class RemBertForCausalLM: ...@@ -3051,6 +3834,9 @@ class RemBertForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RemBertForMaskedLM: class RemBertForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3060,6 +3846,9 @@ class RemBertForMaskedLM: ...@@ -3060,6 +3846,9 @@ class RemBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RemBertForMultipleChoice: class RemBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3069,6 +3858,9 @@ class RemBertForMultipleChoice: ...@@ -3069,6 +3858,9 @@ class RemBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RemBertForQuestionAnswering: class RemBertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3078,6 +3870,9 @@ class RemBertForQuestionAnswering: ...@@ -3078,6 +3870,9 @@ class RemBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RemBertForSequenceClassification: class RemBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3087,6 +3882,9 @@ class RemBertForSequenceClassification: ...@@ -3087,6 +3882,9 @@ class RemBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RemBertForTokenClassification: class RemBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3096,6 +3894,9 @@ class RemBertForTokenClassification: ...@@ -3096,6 +3894,9 @@ class RemBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RemBertLayer: class RemBertLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3110,6 +3911,9 @@ class RemBertModel: ...@@ -3110,6 +3911,9 @@ class RemBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RemBertPreTrainedModel: class RemBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3119,6 +3923,9 @@ class RemBertPreTrainedModel: ...@@ -3119,6 +3923,9 @@ class RemBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_rembert(*args, **kwargs): def load_tf_weights_in_rembert(*args, **kwargs):
requires_backends(load_tf_weights_in_rembert, ["torch"]) requires_backends(load_tf_weights_in_rembert, ["torch"])
...@@ -3135,6 +3942,9 @@ class RetriBertModel: ...@@ -3135,6 +3942,9 @@ class RetriBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RetriBertPreTrainedModel: class RetriBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3144,6 +3954,9 @@ class RetriBertPreTrainedModel: ...@@ -3144,6 +3954,9 @@ class RetriBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3156,6 +3969,9 @@ class RobertaForCausalLM: ...@@ -3156,6 +3969,9 @@ class RobertaForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RobertaForMaskedLM: class RobertaForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3165,6 +3981,9 @@ class RobertaForMaskedLM: ...@@ -3165,6 +3981,9 @@ class RobertaForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RobertaForMultipleChoice: class RobertaForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3174,6 +3993,9 @@ class RobertaForMultipleChoice: ...@@ -3174,6 +3993,9 @@ class RobertaForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RobertaForQuestionAnswering: class RobertaForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3183,6 +4005,9 @@ class RobertaForQuestionAnswering: ...@@ -3183,6 +4005,9 @@ class RobertaForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RobertaForSequenceClassification: class RobertaForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3192,6 +4017,9 @@ class RobertaForSequenceClassification: ...@@ -3192,6 +4017,9 @@ class RobertaForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RobertaForTokenClassification: class RobertaForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3201,6 +4029,9 @@ class RobertaForTokenClassification: ...@@ -3201,6 +4029,9 @@ class RobertaForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RobertaModel: class RobertaModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3210,6 +4041,9 @@ class RobertaModel: ...@@ -3210,6 +4041,9 @@ class RobertaModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RobertaPreTrainedModel: class RobertaPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3219,6 +4053,9 @@ class RobertaPreTrainedModel: ...@@ -3219,6 +4053,9 @@ class RobertaPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
ROFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None ROFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3231,6 +4068,9 @@ class RoFormerForCausalLM: ...@@ -3231,6 +4068,9 @@ class RoFormerForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RoFormerForMaskedLM: class RoFormerForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3240,6 +4080,9 @@ class RoFormerForMaskedLM: ...@@ -3240,6 +4080,9 @@ class RoFormerForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RoFormerForMultipleChoice: class RoFormerForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3249,6 +4092,9 @@ class RoFormerForMultipleChoice: ...@@ -3249,6 +4092,9 @@ class RoFormerForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RoFormerForQuestionAnswering: class RoFormerForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3258,6 +4104,9 @@ class RoFormerForQuestionAnswering: ...@@ -3258,6 +4104,9 @@ class RoFormerForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RoFormerForSequenceClassification: class RoFormerForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3267,6 +4116,9 @@ class RoFormerForSequenceClassification: ...@@ -3267,6 +4116,9 @@ class RoFormerForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RoFormerForTokenClassification: class RoFormerForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3276,6 +4128,9 @@ class RoFormerForTokenClassification: ...@@ -3276,6 +4128,9 @@ class RoFormerForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RoFormerLayer: class RoFormerLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3290,6 +4145,9 @@ class RoFormerModel: ...@@ -3290,6 +4145,9 @@ class RoFormerModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class RoFormerPreTrainedModel: class RoFormerPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3299,6 +4157,9 @@ class RoFormerPreTrainedModel: ...@@ -3299,6 +4157,9 @@ class RoFormerPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_roformer(*args, **kwargs): def load_tf_weights_in_roformer(*args, **kwargs):
requires_backends(load_tf_weights_in_roformer, ["torch"]) requires_backends(load_tf_weights_in_roformer, ["torch"])
...@@ -3335,6 +4196,9 @@ class SegformerModel: ...@@ -3335,6 +4196,9 @@ class SegformerModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SegformerPreTrainedModel: class SegformerPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3344,6 +4208,9 @@ class SegformerPreTrainedModel: ...@@ -3344,6 +4208,9 @@ class SegformerPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
SEW_PRETRAINED_MODEL_ARCHIVE_LIST = None SEW_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3361,6 +4228,9 @@ class SEWForSequenceClassification: ...@@ -3361,6 +4228,9 @@ class SEWForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SEWModel: class SEWModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3370,6 +4240,9 @@ class SEWModel: ...@@ -3370,6 +4240,9 @@ class SEWModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SEWPreTrainedModel: class SEWPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3379,6 +4252,9 @@ class SEWPreTrainedModel: ...@@ -3379,6 +4252,9 @@ class SEWPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
SEW_D_PRETRAINED_MODEL_ARCHIVE_LIST = None SEW_D_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3396,6 +4272,9 @@ class SEWDForSequenceClassification: ...@@ -3396,6 +4272,9 @@ class SEWDForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SEWDModel: class SEWDModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3405,6 +4284,9 @@ class SEWDModel: ...@@ -3405,6 +4284,9 @@ class SEWDModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SEWDPreTrainedModel: class SEWDPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3414,6 +4296,9 @@ class SEWDPreTrainedModel: ...@@ -3414,6 +4296,9 @@ class SEWDPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SpeechEncoderDecoderModel: class SpeechEncoderDecoderModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3423,6 +4308,9 @@ class SpeechEncoderDecoderModel: ...@@ -3423,6 +4308,9 @@ class SpeechEncoderDecoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST = None SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3435,6 +4323,9 @@ class Speech2TextForConditionalGeneration: ...@@ -3435,6 +4323,9 @@ class Speech2TextForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class Speech2TextModel: class Speech2TextModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3444,6 +4335,9 @@ class Speech2TextModel: ...@@ -3444,6 +4335,9 @@ class Speech2TextModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class Speech2TextPreTrainedModel: class Speech2TextPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3453,6 +4347,9 @@ class Speech2TextPreTrainedModel: ...@@ -3453,6 +4347,9 @@ class Speech2TextPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class Speech2Text2ForCausalLM: class Speech2Text2ForCausalLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3462,6 +4359,9 @@ class Speech2Text2ForCausalLM: ...@@ -3462,6 +4359,9 @@ class Speech2Text2ForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class Speech2Text2PreTrainedModel: class Speech2Text2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3471,6 +4371,9 @@ class Speech2Text2PreTrainedModel: ...@@ -3471,6 +4371,9 @@ class Speech2Text2PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
SPLINTER_PRETRAINED_MODEL_ARCHIVE_LIST = None SPLINTER_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3483,6 +4386,9 @@ class SplinterForQuestionAnswering: ...@@ -3483,6 +4386,9 @@ class SplinterForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SplinterLayer: class SplinterLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3497,6 +4403,9 @@ class SplinterModel: ...@@ -3497,6 +4403,9 @@ class SplinterModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SplinterPreTrainedModel: class SplinterPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3506,6 +4415,9 @@ class SplinterPreTrainedModel: ...@@ -3506,6 +4415,9 @@ class SplinterPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
SQUEEZEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None SQUEEZEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3518,6 +4430,9 @@ class SqueezeBertForMaskedLM: ...@@ -3518,6 +4430,9 @@ class SqueezeBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SqueezeBertForMultipleChoice: class SqueezeBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3527,6 +4442,9 @@ class SqueezeBertForMultipleChoice: ...@@ -3527,6 +4442,9 @@ class SqueezeBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SqueezeBertForQuestionAnswering: class SqueezeBertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3536,6 +4454,9 @@ class SqueezeBertForQuestionAnswering: ...@@ -3536,6 +4454,9 @@ class SqueezeBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SqueezeBertForSequenceClassification: class SqueezeBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3545,6 +4466,9 @@ class SqueezeBertForSequenceClassification: ...@@ -3545,6 +4466,9 @@ class SqueezeBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SqueezeBertForTokenClassification: class SqueezeBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3554,6 +4478,9 @@ class SqueezeBertForTokenClassification: ...@@ -3554,6 +4478,9 @@ class SqueezeBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SqueezeBertModel: class SqueezeBertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3563,6 +4490,9 @@ class SqueezeBertModel: ...@@ -3563,6 +4490,9 @@ class SqueezeBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class SqueezeBertModule: class SqueezeBertModule:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3577,6 +4507,9 @@ class SqueezeBertPreTrainedModel: ...@@ -3577,6 +4507,9 @@ class SqueezeBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
T5_PRETRAINED_MODEL_ARCHIVE_LIST = None T5_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3589,6 +4522,9 @@ class T5EncoderModel: ...@@ -3589,6 +4522,9 @@ class T5EncoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class T5ForConditionalGeneration: class T5ForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3598,6 +4534,9 @@ class T5ForConditionalGeneration: ...@@ -3598,6 +4534,9 @@ class T5ForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class T5Model: class T5Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3607,6 +4546,9 @@ class T5Model: ...@@ -3607,6 +4546,9 @@ class T5Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class T5PreTrainedModel: class T5PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3616,6 +4558,9 @@ class T5PreTrainedModel: ...@@ -3616,6 +4558,9 @@ class T5PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_t5(*args, **kwargs): def load_tf_weights_in_t5(*args, **kwargs):
requires_backends(load_tf_weights_in_t5, ["torch"]) requires_backends(load_tf_weights_in_t5, ["torch"])
...@@ -3637,6 +4582,9 @@ class TransfoXLForSequenceClassification: ...@@ -3637,6 +4582,9 @@ class TransfoXLForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class TransfoXLLMHeadModel: class TransfoXLLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3646,6 +4594,9 @@ class TransfoXLLMHeadModel: ...@@ -3646,6 +4594,9 @@ class TransfoXLLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class TransfoXLModel: class TransfoXLModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3655,6 +4606,9 @@ class TransfoXLModel: ...@@ -3655,6 +4606,9 @@ class TransfoXLModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class TransfoXLPreTrainedModel: class TransfoXLPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3664,6 +4618,9 @@ class TransfoXLPreTrainedModel: ...@@ -3664,6 +4618,9 @@ class TransfoXLPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_transfo_xl(*args, **kwargs): def load_tf_weights_in_transfo_xl(*args, **kwargs):
requires_backends(load_tf_weights_in_transfo_xl, ["torch"]) requires_backends(load_tf_weights_in_transfo_xl, ["torch"])
...@@ -3680,6 +4637,9 @@ class TrOCRForCausalLM: ...@@ -3680,6 +4637,9 @@ class TrOCRForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class TrOCRPreTrainedModel: class TrOCRPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3689,6 +4649,9 @@ class TrOCRPreTrainedModel: ...@@ -3689,6 +4649,9 @@ class TrOCRPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
UNISPEECH_PRETRAINED_MODEL_ARCHIVE_LIST = None UNISPEECH_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3711,6 +4674,9 @@ class UniSpeechForSequenceClassification: ...@@ -3711,6 +4674,9 @@ class UniSpeechForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class UniSpeechModel: class UniSpeechModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3720,6 +4686,9 @@ class UniSpeechModel: ...@@ -3720,6 +4686,9 @@ class UniSpeechModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class UniSpeechPreTrainedModel: class UniSpeechPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3729,6 +4698,9 @@ class UniSpeechPreTrainedModel: ...@@ -3729,6 +4698,9 @@ class UniSpeechPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
UNISPEECH_SAT_PRETRAINED_MODEL_ARCHIVE_LIST = None UNISPEECH_SAT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3751,6 +4723,9 @@ class UniSpeechSatForSequenceClassification: ...@@ -3751,6 +4723,9 @@ class UniSpeechSatForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class UniSpeechSatModel: class UniSpeechSatModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3760,6 +4735,9 @@ class UniSpeechSatModel: ...@@ -3760,6 +4735,9 @@ class UniSpeechSatModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class UniSpeechSatPreTrainedModel: class UniSpeechSatPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3769,6 +4747,9 @@ class UniSpeechSatPreTrainedModel: ...@@ -3769,6 +4747,9 @@ class UniSpeechSatPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class VisionEncoderDecoderModel: class VisionEncoderDecoderModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3778,6 +4759,9 @@ class VisionEncoderDecoderModel: ...@@ -3778,6 +4759,9 @@ class VisionEncoderDecoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
VISUAL_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None VISUAL_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3790,6 +4774,9 @@ class VisualBertForMultipleChoice: ...@@ -3790,6 +4774,9 @@ class VisualBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class VisualBertForPreTraining: class VisualBertForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3804,6 +4791,9 @@ class VisualBertForQuestionAnswering: ...@@ -3804,6 +4791,9 @@ class VisualBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class VisualBertForRegionToPhraseAlignment: class VisualBertForRegionToPhraseAlignment:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3828,6 +4818,9 @@ class VisualBertModel: ...@@ -3828,6 +4818,9 @@ class VisualBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class VisualBertPreTrainedModel: class VisualBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3837,6 +4830,9 @@ class VisualBertPreTrainedModel: ...@@ -3837,6 +4830,9 @@ class VisualBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
VIT_PRETRAINED_MODEL_ARCHIVE_LIST = None VIT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3854,6 +4850,9 @@ class ViTModel: ...@@ -3854,6 +4850,9 @@ class ViTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class ViTPreTrainedModel: class ViTPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3863,6 +4862,9 @@ class ViTPreTrainedModel: ...@@ -3863,6 +4862,9 @@ class ViTPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST = None WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3880,6 +4882,9 @@ class Wav2Vec2ForMaskedLM: ...@@ -3880,6 +4882,9 @@ class Wav2Vec2ForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class Wav2Vec2ForPreTraining: class Wav2Vec2ForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3894,6 +4899,9 @@ class Wav2Vec2ForSequenceClassification: ...@@ -3894,6 +4899,9 @@ class Wav2Vec2ForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class Wav2Vec2Model: class Wav2Vec2Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3903,6 +4911,9 @@ class Wav2Vec2Model: ...@@ -3903,6 +4911,9 @@ class Wav2Vec2Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class Wav2Vec2PreTrainedModel: class Wav2Vec2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3912,6 +4923,9 @@ class Wav2Vec2PreTrainedModel: ...@@ -3912,6 +4923,9 @@ class Wav2Vec2PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -3924,6 +4938,9 @@ class XLMForMultipleChoice: ...@@ -3924,6 +4938,9 @@ class XLMForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMForQuestionAnswering: class XLMForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3933,6 +4950,9 @@ class XLMForQuestionAnswering: ...@@ -3933,6 +4950,9 @@ class XLMForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMForQuestionAnsweringSimple: class XLMForQuestionAnsweringSimple:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3942,6 +4962,9 @@ class XLMForQuestionAnsweringSimple: ...@@ -3942,6 +4962,9 @@ class XLMForQuestionAnsweringSimple:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMForSequenceClassification: class XLMForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3951,6 +4974,9 @@ class XLMForSequenceClassification: ...@@ -3951,6 +4974,9 @@ class XLMForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMForTokenClassification: class XLMForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3960,6 +4986,9 @@ class XLMForTokenClassification: ...@@ -3960,6 +4986,9 @@ class XLMForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMModel: class XLMModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3969,6 +4998,9 @@ class XLMModel: ...@@ -3969,6 +4998,9 @@ class XLMModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMPreTrainedModel: class XLMPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3978,6 +5010,9 @@ class XLMPreTrainedModel: ...@@ -3978,6 +5010,9 @@ class XLMPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMWithLMHeadModel: class XLMWithLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -3987,6 +5022,9 @@ class XLMWithLMHeadModel: ...@@ -3987,6 +5022,9 @@ class XLMWithLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
XLM_PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LIST = None XLM_PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -4009,6 +5047,9 @@ class XLMProphetNetForCausalLM: ...@@ -4009,6 +5047,9 @@ class XLMProphetNetForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMProphetNetForConditionalGeneration: class XLMProphetNetForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4018,6 +5059,9 @@ class XLMProphetNetForConditionalGeneration: ...@@ -4018,6 +5059,9 @@ class XLMProphetNetForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMProphetNetModel: class XLMProphetNetModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4027,6 +5071,9 @@ class XLMProphetNetModel: ...@@ -4027,6 +5071,9 @@ class XLMProphetNetModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -4039,6 +5086,9 @@ class XLMRobertaForCausalLM: ...@@ -4039,6 +5086,9 @@ class XLMRobertaForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMRobertaForMaskedLM: class XLMRobertaForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4048,6 +5098,9 @@ class XLMRobertaForMaskedLM: ...@@ -4048,6 +5098,9 @@ class XLMRobertaForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMRobertaForMultipleChoice: class XLMRobertaForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4057,6 +5110,9 @@ class XLMRobertaForMultipleChoice: ...@@ -4057,6 +5110,9 @@ class XLMRobertaForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMRobertaForQuestionAnswering: class XLMRobertaForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4066,6 +5122,9 @@ class XLMRobertaForQuestionAnswering: ...@@ -4066,6 +5122,9 @@ class XLMRobertaForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMRobertaForSequenceClassification: class XLMRobertaForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4075,6 +5134,9 @@ class XLMRobertaForSequenceClassification: ...@@ -4075,6 +5134,9 @@ class XLMRobertaForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMRobertaForTokenClassification: class XLMRobertaForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4084,6 +5146,9 @@ class XLMRobertaForTokenClassification: ...@@ -4084,6 +5146,9 @@ class XLMRobertaForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLMRobertaModel: class XLMRobertaModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4093,6 +5158,9 @@ class XLMRobertaModel: ...@@ -4093,6 +5158,9 @@ class XLMRobertaModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
XLNET_PRETRAINED_MODEL_ARCHIVE_LIST = None XLNET_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -4105,6 +5173,9 @@ class XLNetForMultipleChoice: ...@@ -4105,6 +5173,9 @@ class XLNetForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLNetForQuestionAnswering: class XLNetForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4114,6 +5185,9 @@ class XLNetForQuestionAnswering: ...@@ -4114,6 +5185,9 @@ class XLNetForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLNetForQuestionAnsweringSimple: class XLNetForQuestionAnsweringSimple:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4123,6 +5197,9 @@ class XLNetForQuestionAnsweringSimple: ...@@ -4123,6 +5197,9 @@ class XLNetForQuestionAnsweringSimple:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLNetForSequenceClassification: class XLNetForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4132,6 +5209,9 @@ class XLNetForSequenceClassification: ...@@ -4132,6 +5209,9 @@ class XLNetForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLNetForTokenClassification: class XLNetForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4141,6 +5221,9 @@ class XLNetForTokenClassification: ...@@ -4141,6 +5221,9 @@ class XLNetForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLNetLMHeadModel: class XLNetLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4150,6 +5233,9 @@ class XLNetLMHeadModel: ...@@ -4150,6 +5233,9 @@ class XLNetLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLNetModel: class XLNetModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4159,6 +5245,9 @@ class XLNetModel: ...@@ -4159,6 +5245,9 @@ class XLNetModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
class XLNetPreTrainedModel: class XLNetPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -4168,6 +5257,9 @@ class XLNetPreTrainedModel: ...@@ -4168,6 +5257,9 @@ class XLNetPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"]) requires_backends(cls, ["torch"])
def forward(self, *args, **kwargs):
requires_backends(self, ["torch"])
def load_tf_weights_in_xlnet(*args, **kwargs): def load_tf_weights_in_xlnet(*args, **kwargs):
requires_backends(load_tf_weights_in_xlnet, ["torch"]) requires_backends(load_tf_weights_in_xlnet, ["torch"])
......
src/transformers/utils/dummy_scatter_objects.py
View file @ 3e8d17e6
...@@ -13,6 +13,9 @@ class TapasForMaskedLM: ...@@ -13,6 +13,9 @@ class TapasForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["scatter"]) requires_backends(cls, ["scatter"])
def forward(self, *args, **kwargs):
requires_backends(self, ["scatter"])
class TapasForQuestionAnswering: class TapasForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -22,6 +25,9 @@ class TapasForQuestionAnswering: ...@@ -22,6 +25,9 @@ class TapasForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["scatter"]) requires_backends(cls, ["scatter"])
def forward(self, *args, **kwargs):
requires_backends(self, ["scatter"])
class TapasForSequenceClassification: class TapasForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -31,6 +37,9 @@ class TapasForSequenceClassification: ...@@ -31,6 +37,9 @@ class TapasForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["scatter"]) requires_backends(cls, ["scatter"])
def forward(self, *args, **kwargs):
requires_backends(self, ["scatter"])
class TapasModel: class TapasModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -40,6 +49,9 @@ class TapasModel: ...@@ -40,6 +49,9 @@ class TapasModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["scatter"]) requires_backends(cls, ["scatter"])
def forward(self, *args, **kwargs):
requires_backends(self, ["scatter"])
class TapasPreTrainedModel: class TapasPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -49,6 +61,9 @@ class TapasPreTrainedModel: ...@@ -49,6 +61,9 @@ class TapasPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["scatter"]) requires_backends(cls, ["scatter"])
def forward(self, *args, **kwargs):
requires_backends(self, ["scatter"])
def load_tf_weights_in_tapas(*args, **kwargs): def load_tf_weights_in_tapas(*args, **kwargs):
requires_backends(load_tf_weights_in_tapas, ["scatter"]) requires_backends(load_tf_weights_in_tapas, ["scatter"])
src/transformers/utils/dummy_tf_objects.py
View file @ 3e8d17e6
...@@ -32,6 +32,9 @@ class TFLayoutLMForMaskedLM: ...@@ -32,6 +32,9 @@ class TFLayoutLMForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLayoutLMForSequenceClassification: class TFLayoutLMForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -41,6 +44,9 @@ class TFLayoutLMForSequenceClassification: ...@@ -41,6 +44,9 @@ class TFLayoutLMForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLayoutLMForTokenClassification: class TFLayoutLMForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -50,6 +56,9 @@ class TFLayoutLMForTokenClassification: ...@@ -50,6 +56,9 @@ class TFLayoutLMForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLayoutLMMainLayer: class TFLayoutLMMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -64,6 +73,9 @@ class TFLayoutLMModel: ...@@ -64,6 +73,9 @@ class TFLayoutLMModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLayoutLMPreTrainedModel: class TFLayoutLMPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -73,6 +85,9 @@ class TFLayoutLMPreTrainedModel: ...@@ -73,6 +85,9 @@ class TFLayoutLMPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFPreTrainedModel: class TFPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -82,6 +97,9 @@ class TFPreTrainedModel: ...@@ -82,6 +97,9 @@ class TFPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFSequenceSummary: class TFSequenceSummary:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -108,6 +126,9 @@ class TFAlbertForMaskedLM: ...@@ -108,6 +126,9 @@ class TFAlbertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAlbertForMultipleChoice: class TFAlbertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -117,6 +138,9 @@ class TFAlbertForMultipleChoice: ...@@ -117,6 +138,9 @@ class TFAlbertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAlbertForPreTraining: class TFAlbertForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -131,6 +155,9 @@ class TFAlbertForQuestionAnswering: ...@@ -131,6 +155,9 @@ class TFAlbertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAlbertForSequenceClassification: class TFAlbertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -140,6 +167,9 @@ class TFAlbertForSequenceClassification: ...@@ -140,6 +167,9 @@ class TFAlbertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAlbertForTokenClassification: class TFAlbertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -149,6 +179,9 @@ class TFAlbertForTokenClassification: ...@@ -149,6 +179,9 @@ class TFAlbertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAlbertMainLayer: class TFAlbertMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -163,6 +196,9 @@ class TFAlbertModel: ...@@ -163,6 +196,9 @@ class TFAlbertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAlbertPreTrainedModel: class TFAlbertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -172,6 +208,9 @@ class TFAlbertPreTrainedModel: ...@@ -172,6 +208,9 @@ class TFAlbertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_MODEL_FOR_CAUSAL_LM_MAPPING = None TF_MODEL_FOR_CAUSAL_LM_MAPPING = None
...@@ -217,6 +256,9 @@ class TFAutoModel: ...@@ -217,6 +256,9 @@ class TFAutoModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForCausalLM: class TFAutoModelForCausalLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -226,6 +268,9 @@ class TFAutoModelForCausalLM: ...@@ -226,6 +268,9 @@ class TFAutoModelForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForImageClassification: class TFAutoModelForImageClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -235,6 +280,9 @@ class TFAutoModelForImageClassification: ...@@ -235,6 +280,9 @@ class TFAutoModelForImageClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForMaskedLM: class TFAutoModelForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -244,6 +292,9 @@ class TFAutoModelForMaskedLM: ...@@ -244,6 +292,9 @@ class TFAutoModelForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForMultipleChoice: class TFAutoModelForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -253,6 +304,9 @@ class TFAutoModelForMultipleChoice: ...@@ -253,6 +304,9 @@ class TFAutoModelForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForPreTraining: class TFAutoModelForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -262,6 +316,9 @@ class TFAutoModelForPreTraining: ...@@ -262,6 +316,9 @@ class TFAutoModelForPreTraining:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForQuestionAnswering: class TFAutoModelForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -271,6 +328,9 @@ class TFAutoModelForQuestionAnswering: ...@@ -271,6 +328,9 @@ class TFAutoModelForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForSeq2SeqLM: class TFAutoModelForSeq2SeqLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -280,6 +340,9 @@ class TFAutoModelForSeq2SeqLM: ...@@ -280,6 +340,9 @@ class TFAutoModelForSeq2SeqLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForSequenceClassification: class TFAutoModelForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -289,6 +352,9 @@ class TFAutoModelForSequenceClassification: ...@@ -289,6 +352,9 @@ class TFAutoModelForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForTokenClassification: class TFAutoModelForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -298,6 +364,9 @@ class TFAutoModelForTokenClassification: ...@@ -298,6 +364,9 @@ class TFAutoModelForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelWithLMHead: class TFAutoModelWithLMHead:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -307,6 +376,9 @@ class TFAutoModelWithLMHead: ...@@ -307,6 +376,9 @@ class TFAutoModelWithLMHead:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBartForConditionalGeneration: class TFBartForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -316,6 +388,9 @@ class TFBartForConditionalGeneration: ...@@ -316,6 +388,9 @@ class TFBartForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBartModel: class TFBartModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -325,6 +400,9 @@ class TFBartModel: ...@@ -325,6 +400,9 @@ class TFBartModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBartPretrainedModel: class TFBartPretrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -334,6 +412,9 @@ class TFBartPretrainedModel: ...@@ -334,6 +412,9 @@ class TFBartPretrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -351,6 +432,9 @@ class TFBertForMaskedLM: ...@@ -351,6 +432,9 @@ class TFBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBertForMultipleChoice: class TFBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -360,6 +444,9 @@ class TFBertForMultipleChoice: ...@@ -360,6 +444,9 @@ class TFBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBertForNextSentencePrediction: class TFBertForNextSentencePrediction:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -379,6 +466,9 @@ class TFBertForQuestionAnswering: ...@@ -379,6 +466,9 @@ class TFBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBertForSequenceClassification: class TFBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -388,6 +478,9 @@ class TFBertForSequenceClassification: ...@@ -388,6 +478,9 @@ class TFBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBertForTokenClassification: class TFBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -397,6 +490,9 @@ class TFBertForTokenClassification: ...@@ -397,6 +490,9 @@ class TFBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBertLMHeadModel: class TFBertLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -406,6 +502,9 @@ class TFBertLMHeadModel: ...@@ -406,6 +502,9 @@ class TFBertLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBertMainLayer: class TFBertMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -420,6 +519,9 @@ class TFBertModel: ...@@ -420,6 +519,9 @@ class TFBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBertPreTrainedModel: class TFBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -429,6 +531,9 @@ class TFBertPreTrainedModel: ...@@ -429,6 +531,9 @@ class TFBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBlenderbotForConditionalGeneration: class TFBlenderbotForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -438,6 +543,9 @@ class TFBlenderbotForConditionalGeneration: ...@@ -438,6 +543,9 @@ class TFBlenderbotForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBlenderbotModel: class TFBlenderbotModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -447,6 +555,9 @@ class TFBlenderbotModel: ...@@ -447,6 +555,9 @@ class TFBlenderbotModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBlenderbotPreTrainedModel: class TFBlenderbotPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -456,6 +567,9 @@ class TFBlenderbotPreTrainedModel: ...@@ -456,6 +567,9 @@ class TFBlenderbotPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBlenderbotSmallForConditionalGeneration: class TFBlenderbotSmallForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -465,6 +579,9 @@ class TFBlenderbotSmallForConditionalGeneration: ...@@ -465,6 +579,9 @@ class TFBlenderbotSmallForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBlenderbotSmallModel: class TFBlenderbotSmallModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -474,6 +591,9 @@ class TFBlenderbotSmallModel: ...@@ -474,6 +591,9 @@ class TFBlenderbotSmallModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFBlenderbotSmallPreTrainedModel: class TFBlenderbotSmallPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -483,6 +603,9 @@ class TFBlenderbotSmallPreTrainedModel: ...@@ -483,6 +603,9 @@ class TFBlenderbotSmallPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -495,6 +618,9 @@ class TFCamembertForMaskedLM: ...@@ -495,6 +618,9 @@ class TFCamembertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFCamembertForMultipleChoice: class TFCamembertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -504,6 +630,9 @@ class TFCamembertForMultipleChoice: ...@@ -504,6 +630,9 @@ class TFCamembertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFCamembertForQuestionAnswering: class TFCamembertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -513,6 +642,9 @@ class TFCamembertForQuestionAnswering: ...@@ -513,6 +642,9 @@ class TFCamembertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFCamembertForSequenceClassification: class TFCamembertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -522,6 +654,9 @@ class TFCamembertForSequenceClassification: ...@@ -522,6 +654,9 @@ class TFCamembertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFCamembertForTokenClassification: class TFCamembertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -531,6 +666,9 @@ class TFCamembertForTokenClassification: ...@@ -531,6 +666,9 @@ class TFCamembertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFCamembertModel: class TFCamembertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -540,6 +678,9 @@ class TFCamembertModel: ...@@ -540,6 +678,9 @@ class TFCamembertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_CONVBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_CONVBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -552,6 +693,9 @@ class TFConvBertForMaskedLM: ...@@ -552,6 +693,9 @@ class TFConvBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFConvBertForMultipleChoice: class TFConvBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -561,6 +705,9 @@ class TFConvBertForMultipleChoice: ...@@ -561,6 +705,9 @@ class TFConvBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFConvBertForQuestionAnswering: class TFConvBertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -570,6 +717,9 @@ class TFConvBertForQuestionAnswering: ...@@ -570,6 +717,9 @@ class TFConvBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFConvBertForSequenceClassification: class TFConvBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -579,6 +729,9 @@ class TFConvBertForSequenceClassification: ...@@ -579,6 +729,9 @@ class TFConvBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFConvBertForTokenClassification: class TFConvBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -588,6 +741,9 @@ class TFConvBertForTokenClassification: ...@@ -588,6 +741,9 @@ class TFConvBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFConvBertLayer: class TFConvBertLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -602,6 +758,9 @@ class TFConvBertModel: ...@@ -602,6 +758,9 @@ class TFConvBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFConvBertPreTrainedModel: class TFConvBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -611,6 +770,9 @@ class TFConvBertPreTrainedModel: ...@@ -611,6 +770,9 @@ class TFConvBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_CTRL_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_CTRL_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -623,6 +785,9 @@ class TFCTRLForSequenceClassification: ...@@ -623,6 +785,9 @@ class TFCTRLForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFCTRLLMHeadModel: class TFCTRLLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -632,6 +797,9 @@ class TFCTRLLMHeadModel: ...@@ -632,6 +797,9 @@ class TFCTRLLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFCTRLModel: class TFCTRLModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -641,6 +809,9 @@ class TFCTRLModel: ...@@ -641,6 +809,9 @@ class TFCTRLModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFCTRLPreTrainedModel: class TFCTRLPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -650,6 +821,9 @@ class TFCTRLPreTrainedModel: ...@@ -650,6 +821,9 @@ class TFCTRLPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_DEBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_DEBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -662,6 +836,9 @@ class TFDebertaForMaskedLM: ...@@ -662,6 +836,9 @@ class TFDebertaForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaForQuestionAnswering: class TFDebertaForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -671,6 +848,9 @@ class TFDebertaForQuestionAnswering: ...@@ -671,6 +848,9 @@ class TFDebertaForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaForSequenceClassification: class TFDebertaForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -680,6 +860,9 @@ class TFDebertaForSequenceClassification: ...@@ -680,6 +860,9 @@ class TFDebertaForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaForTokenClassification: class TFDebertaForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -689,6 +872,9 @@ class TFDebertaForTokenClassification: ...@@ -689,6 +872,9 @@ class TFDebertaForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaModel: class TFDebertaModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -698,6 +884,9 @@ class TFDebertaModel: ...@@ -698,6 +884,9 @@ class TFDebertaModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaPreTrainedModel: class TFDebertaPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -707,6 +896,9 @@ class TFDebertaPreTrainedModel: ...@@ -707,6 +896,9 @@ class TFDebertaPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_DEBERTA_V2_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_DEBERTA_V2_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -719,6 +911,9 @@ class TFDebertaV2ForMaskedLM: ...@@ -719,6 +911,9 @@ class TFDebertaV2ForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaV2ForQuestionAnswering: class TFDebertaV2ForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -728,6 +923,9 @@ class TFDebertaV2ForQuestionAnswering: ...@@ -728,6 +923,9 @@ class TFDebertaV2ForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaV2ForSequenceClassification: class TFDebertaV2ForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -737,6 +935,9 @@ class TFDebertaV2ForSequenceClassification: ...@@ -737,6 +935,9 @@ class TFDebertaV2ForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaV2ForTokenClassification: class TFDebertaV2ForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -746,6 +947,9 @@ class TFDebertaV2ForTokenClassification: ...@@ -746,6 +947,9 @@ class TFDebertaV2ForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaV2Model: class TFDebertaV2Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -755,6 +959,9 @@ class TFDebertaV2Model: ...@@ -755,6 +959,9 @@ class TFDebertaV2Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDebertaV2PreTrainedModel: class TFDebertaV2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -764,6 +971,9 @@ class TFDebertaV2PreTrainedModel: ...@@ -764,6 +971,9 @@ class TFDebertaV2PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -776,6 +986,9 @@ class TFDistilBertForMaskedLM: ...@@ -776,6 +986,9 @@ class TFDistilBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDistilBertForMultipleChoice: class TFDistilBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -785,6 +998,9 @@ class TFDistilBertForMultipleChoice: ...@@ -785,6 +998,9 @@ class TFDistilBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDistilBertForQuestionAnswering: class TFDistilBertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -794,6 +1010,9 @@ class TFDistilBertForQuestionAnswering: ...@@ -794,6 +1010,9 @@ class TFDistilBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDistilBertForSequenceClassification: class TFDistilBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -803,6 +1022,9 @@ class TFDistilBertForSequenceClassification: ...@@ -803,6 +1022,9 @@ class TFDistilBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDistilBertForTokenClassification: class TFDistilBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -812,6 +1034,9 @@ class TFDistilBertForTokenClassification: ...@@ -812,6 +1034,9 @@ class TFDistilBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDistilBertMainLayer: class TFDistilBertMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -826,6 +1051,9 @@ class TFDistilBertModel: ...@@ -826,6 +1051,9 @@ class TFDistilBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFDistilBertPreTrainedModel: class TFDistilBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -835,6 +1063,9 @@ class TFDistilBertPreTrainedModel: ...@@ -835,6 +1063,9 @@ class TFDistilBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -886,6 +1117,9 @@ class TFElectraForMaskedLM: ...@@ -886,6 +1117,9 @@ class TFElectraForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFElectraForMultipleChoice: class TFElectraForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -895,6 +1129,9 @@ class TFElectraForMultipleChoice: ...@@ -895,6 +1129,9 @@ class TFElectraForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFElectraForPreTraining: class TFElectraForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -909,6 +1146,9 @@ class TFElectraForQuestionAnswering: ...@@ -909,6 +1146,9 @@ class TFElectraForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFElectraForSequenceClassification: class TFElectraForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -918,6 +1158,9 @@ class TFElectraForSequenceClassification: ...@@ -918,6 +1158,9 @@ class TFElectraForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFElectraForTokenClassification: class TFElectraForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -927,6 +1170,9 @@ class TFElectraForTokenClassification: ...@@ -927,6 +1170,9 @@ class TFElectraForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFElectraModel: class TFElectraModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -936,6 +1182,9 @@ class TFElectraModel: ...@@ -936,6 +1182,9 @@ class TFElectraModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFElectraPreTrainedModel: class TFElectraPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -945,6 +1194,9 @@ class TFElectraPreTrainedModel: ...@@ -945,6 +1194,9 @@ class TFElectraPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFEncoderDecoderModel: class TFEncoderDecoderModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -954,6 +1206,9 @@ class TFEncoderDecoderModel: ...@@ -954,6 +1206,9 @@ class TFEncoderDecoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -966,6 +1221,9 @@ class TFFlaubertForMultipleChoice: ...@@ -966,6 +1221,9 @@ class TFFlaubertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFlaubertForQuestionAnsweringSimple: class TFFlaubertForQuestionAnsweringSimple:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -975,6 +1233,9 @@ class TFFlaubertForQuestionAnsweringSimple: ...@@ -975,6 +1233,9 @@ class TFFlaubertForQuestionAnsweringSimple:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFlaubertForSequenceClassification: class TFFlaubertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -984,6 +1245,9 @@ class TFFlaubertForSequenceClassification: ...@@ -984,6 +1245,9 @@ class TFFlaubertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFlaubertForTokenClassification: class TFFlaubertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -993,6 +1257,9 @@ class TFFlaubertForTokenClassification: ...@@ -993,6 +1257,9 @@ class TFFlaubertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFlaubertModel: class TFFlaubertModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1002,6 +1269,9 @@ class TFFlaubertModel: ...@@ -1002,6 +1269,9 @@ class TFFlaubertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFlaubertPreTrainedModel: class TFFlaubertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1011,6 +1281,9 @@ class TFFlaubertPreTrainedModel: ...@@ -1011,6 +1281,9 @@ class TFFlaubertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFlaubertWithLMHeadModel: class TFFlaubertWithLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1020,6 +1293,9 @@ class TFFlaubertWithLMHeadModel: ...@@ -1020,6 +1293,9 @@ class TFFlaubertWithLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_FUNNEL_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_FUNNEL_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1032,6 +1308,9 @@ class TFFunnelBaseModel: ...@@ -1032,6 +1308,9 @@ class TFFunnelBaseModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFunnelForMaskedLM: class TFFunnelForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1041,6 +1320,9 @@ class TFFunnelForMaskedLM: ...@@ -1041,6 +1320,9 @@ class TFFunnelForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFunnelForMultipleChoice: class TFFunnelForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1050,6 +1332,9 @@ class TFFunnelForMultipleChoice: ...@@ -1050,6 +1332,9 @@ class TFFunnelForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFunnelForPreTraining: class TFFunnelForPreTraining:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1064,6 +1349,9 @@ class TFFunnelForQuestionAnswering: ...@@ -1064,6 +1349,9 @@ class TFFunnelForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFunnelForSequenceClassification: class TFFunnelForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1073,6 +1361,9 @@ class TFFunnelForSequenceClassification: ...@@ -1073,6 +1361,9 @@ class TFFunnelForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFunnelForTokenClassification: class TFFunnelForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1082,6 +1373,9 @@ class TFFunnelForTokenClassification: ...@@ -1082,6 +1373,9 @@ class TFFunnelForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFunnelModel: class TFFunnelModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1091,6 +1385,9 @@ class TFFunnelModel: ...@@ -1091,6 +1385,9 @@ class TFFunnelModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFFunnelPreTrainedModel: class TFFunnelPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1100,6 +1397,9 @@ class TFFunnelPreTrainedModel: ...@@ -1100,6 +1397,9 @@ class TFFunnelPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1112,6 +1412,9 @@ class TFGPT2DoubleHeadsModel: ...@@ -1112,6 +1412,9 @@ class TFGPT2DoubleHeadsModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFGPT2ForSequenceClassification: class TFGPT2ForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1121,6 +1424,9 @@ class TFGPT2ForSequenceClassification: ...@@ -1121,6 +1424,9 @@ class TFGPT2ForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFGPT2LMHeadModel: class TFGPT2LMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1130,6 +1436,9 @@ class TFGPT2LMHeadModel: ...@@ -1130,6 +1436,9 @@ class TFGPT2LMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFGPT2MainLayer: class TFGPT2MainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1144,6 +1453,9 @@ class TFGPT2Model: ...@@ -1144,6 +1453,9 @@ class TFGPT2Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFGPT2PreTrainedModel: class TFGPT2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1153,6 +1465,9 @@ class TFGPT2PreTrainedModel: ...@@ -1153,6 +1465,9 @@ class TFGPT2PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1170,6 +1485,9 @@ class TFHubertModel: ...@@ -1170,6 +1485,9 @@ class TFHubertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFHubertPreTrainedModel: class TFHubertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1179,6 +1497,9 @@ class TFHubertPreTrainedModel: ...@@ -1179,6 +1497,9 @@ class TFHubertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLEDForConditionalGeneration: class TFLEDForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1188,6 +1509,9 @@ class TFLEDForConditionalGeneration: ...@@ -1188,6 +1509,9 @@ class TFLEDForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLEDModel: class TFLEDModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1197,6 +1521,9 @@ class TFLEDModel: ...@@ -1197,6 +1521,9 @@ class TFLEDModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLEDPreTrainedModel: class TFLEDPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1206,6 +1533,9 @@ class TFLEDPreTrainedModel: ...@@ -1206,6 +1533,9 @@ class TFLEDPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1218,6 +1548,9 @@ class TFLongformerForMaskedLM: ...@@ -1218,6 +1548,9 @@ class TFLongformerForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLongformerForMultipleChoice: class TFLongformerForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1227,6 +1560,9 @@ class TFLongformerForMultipleChoice: ...@@ -1227,6 +1560,9 @@ class TFLongformerForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLongformerForQuestionAnswering: class TFLongformerForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1236,6 +1572,9 @@ class TFLongformerForQuestionAnswering: ...@@ -1236,6 +1572,9 @@ class TFLongformerForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLongformerForSequenceClassification: class TFLongformerForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1245,6 +1584,9 @@ class TFLongformerForSequenceClassification: ...@@ -1245,6 +1584,9 @@ class TFLongformerForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLongformerForTokenClassification: class TFLongformerForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1254,6 +1596,9 @@ class TFLongformerForTokenClassification: ...@@ -1254,6 +1596,9 @@ class TFLongformerForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLongformerModel: class TFLongformerModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1263,6 +1608,9 @@ class TFLongformerModel: ...@@ -1263,6 +1608,9 @@ class TFLongformerModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLongformerPreTrainedModel: class TFLongformerPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1272,6 +1620,9 @@ class TFLongformerPreTrainedModel: ...@@ -1272,6 +1620,9 @@ class TFLongformerPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLongformerSelfAttention: class TFLongformerSelfAttention:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1299,6 +1650,9 @@ class TFLxmertModel: ...@@ -1299,6 +1650,9 @@ class TFLxmertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLxmertPreTrainedModel: class TFLxmertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1308,6 +1662,9 @@ class TFLxmertPreTrainedModel: ...@@ -1308,6 +1662,9 @@ class TFLxmertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFLxmertVisualFeatureEncoder: class TFLxmertVisualFeatureEncoder:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1322,6 +1679,9 @@ class TFMarianModel: ...@@ -1322,6 +1679,9 @@ class TFMarianModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMarianMTModel: class TFMarianMTModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1331,6 +1691,9 @@ class TFMarianMTModel: ...@@ -1331,6 +1691,9 @@ class TFMarianMTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMarianPreTrainedModel: class TFMarianPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1340,6 +1703,9 @@ class TFMarianPreTrainedModel: ...@@ -1340,6 +1703,9 @@ class TFMarianPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMBartForConditionalGeneration: class TFMBartForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1349,6 +1715,9 @@ class TFMBartForConditionalGeneration: ...@@ -1349,6 +1715,9 @@ class TFMBartForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMBartModel: class TFMBartModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1358,6 +1727,9 @@ class TFMBartModel: ...@@ -1358,6 +1727,9 @@ class TFMBartModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMBartPreTrainedModel: class TFMBartPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1367,6 +1739,9 @@ class TFMBartPreTrainedModel: ...@@ -1367,6 +1739,9 @@ class TFMBartPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1379,6 +1754,9 @@ class TFMobileBertForMaskedLM: ...@@ -1379,6 +1754,9 @@ class TFMobileBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMobileBertForMultipleChoice: class TFMobileBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1388,6 +1766,9 @@ class TFMobileBertForMultipleChoice: ...@@ -1388,6 +1766,9 @@ class TFMobileBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMobileBertForNextSentencePrediction: class TFMobileBertForNextSentencePrediction:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1407,6 +1788,9 @@ class TFMobileBertForQuestionAnswering: ...@@ -1407,6 +1788,9 @@ class TFMobileBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMobileBertForSequenceClassification: class TFMobileBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1416,6 +1800,9 @@ class TFMobileBertForSequenceClassification: ...@@ -1416,6 +1800,9 @@ class TFMobileBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMobileBertForTokenClassification: class TFMobileBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1425,6 +1812,9 @@ class TFMobileBertForTokenClassification: ...@@ -1425,6 +1812,9 @@ class TFMobileBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMobileBertMainLayer: class TFMobileBertMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1439,6 +1829,9 @@ class TFMobileBertModel: ...@@ -1439,6 +1829,9 @@ class TFMobileBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMobileBertPreTrainedModel: class TFMobileBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1448,6 +1841,9 @@ class TFMobileBertPreTrainedModel: ...@@ -1448,6 +1841,9 @@ class TFMobileBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_MPNET_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_MPNET_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1460,6 +1856,9 @@ class TFMPNetForMaskedLM: ...@@ -1460,6 +1856,9 @@ class TFMPNetForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMPNetForMultipleChoice: class TFMPNetForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1469,6 +1868,9 @@ class TFMPNetForMultipleChoice: ...@@ -1469,6 +1868,9 @@ class TFMPNetForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMPNetForQuestionAnswering: class TFMPNetForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1478,6 +1880,9 @@ class TFMPNetForQuestionAnswering: ...@@ -1478,6 +1880,9 @@ class TFMPNetForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMPNetForSequenceClassification: class TFMPNetForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1487,6 +1892,9 @@ class TFMPNetForSequenceClassification: ...@@ -1487,6 +1892,9 @@ class TFMPNetForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMPNetForTokenClassification: class TFMPNetForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1496,6 +1904,9 @@ class TFMPNetForTokenClassification: ...@@ -1496,6 +1904,9 @@ class TFMPNetForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMPNetMainLayer: class TFMPNetMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1510,6 +1921,9 @@ class TFMPNetModel: ...@@ -1510,6 +1921,9 @@ class TFMPNetModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMPNetPreTrainedModel: class TFMPNetPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1519,6 +1933,9 @@ class TFMPNetPreTrainedModel: ...@@ -1519,6 +1933,9 @@ class TFMPNetPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMT5EncoderModel: class TFMT5EncoderModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1528,6 +1945,9 @@ class TFMT5EncoderModel: ...@@ -1528,6 +1945,9 @@ class TFMT5EncoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMT5ForConditionalGeneration: class TFMT5ForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1537,6 +1957,9 @@ class TFMT5ForConditionalGeneration: ...@@ -1537,6 +1957,9 @@ class TFMT5ForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFMT5Model: class TFMT5Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1546,6 +1969,9 @@ class TFMT5Model: ...@@ -1546,6 +1969,9 @@ class TFMT5Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1558,6 +1984,9 @@ class TFOpenAIGPTDoubleHeadsModel: ...@@ -1558,6 +1984,9 @@ class TFOpenAIGPTDoubleHeadsModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFOpenAIGPTForSequenceClassification: class TFOpenAIGPTForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1567,6 +1996,9 @@ class TFOpenAIGPTForSequenceClassification: ...@@ -1567,6 +1996,9 @@ class TFOpenAIGPTForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFOpenAIGPTLMHeadModel: class TFOpenAIGPTLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1576,6 +2008,9 @@ class TFOpenAIGPTLMHeadModel: ...@@ -1576,6 +2008,9 @@ class TFOpenAIGPTLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFOpenAIGPTMainLayer: class TFOpenAIGPTMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1590,6 +2025,9 @@ class TFOpenAIGPTModel: ...@@ -1590,6 +2025,9 @@ class TFOpenAIGPTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFOpenAIGPTPreTrainedModel: class TFOpenAIGPTPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1599,6 +2037,9 @@ class TFOpenAIGPTPreTrainedModel: ...@@ -1599,6 +2037,9 @@ class TFOpenAIGPTPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFPegasusForConditionalGeneration: class TFPegasusForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1608,6 +2049,9 @@ class TFPegasusForConditionalGeneration: ...@@ -1608,6 +2049,9 @@ class TFPegasusForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFPegasusModel: class TFPegasusModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1617,6 +2061,9 @@ class TFPegasusModel: ...@@ -1617,6 +2061,9 @@ class TFPegasusModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFPegasusPreTrainedModel: class TFPegasusPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1626,6 +2073,9 @@ class TFPegasusPreTrainedModel: ...@@ -1626,6 +2073,9 @@ class TFPegasusPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRagModel: class TFRagModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1635,6 +2085,9 @@ class TFRagModel: ...@@ -1635,6 +2085,9 @@ class TFRagModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRagPreTrainedModel: class TFRagPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1644,6 +2097,9 @@ class TFRagPreTrainedModel: ...@@ -1644,6 +2097,9 @@ class TFRagPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRagSequenceForGeneration: class TFRagSequenceForGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1666,6 +2122,9 @@ class TFRemBertForCausalLM: ...@@ -1666,6 +2122,9 @@ class TFRemBertForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRemBertForMaskedLM: class TFRemBertForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1675,6 +2134,9 @@ class TFRemBertForMaskedLM: ...@@ -1675,6 +2134,9 @@ class TFRemBertForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRemBertForMultipleChoice: class TFRemBertForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1684,6 +2146,9 @@ class TFRemBertForMultipleChoice: ...@@ -1684,6 +2146,9 @@ class TFRemBertForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRemBertForQuestionAnswering: class TFRemBertForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1693,6 +2158,9 @@ class TFRemBertForQuestionAnswering: ...@@ -1693,6 +2158,9 @@ class TFRemBertForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRemBertForSequenceClassification: class TFRemBertForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1702,6 +2170,9 @@ class TFRemBertForSequenceClassification: ...@@ -1702,6 +2170,9 @@ class TFRemBertForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRemBertForTokenClassification: class TFRemBertForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1711,6 +2182,9 @@ class TFRemBertForTokenClassification: ...@@ -1711,6 +2182,9 @@ class TFRemBertForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRemBertLayer: class TFRemBertLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1725,6 +2199,9 @@ class TFRemBertModel: ...@@ -1725,6 +2199,9 @@ class TFRemBertModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRemBertPreTrainedModel: class TFRemBertPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1734,6 +2211,9 @@ class TFRemBertPreTrainedModel: ...@@ -1734,6 +2211,9 @@ class TFRemBertPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1746,6 +2226,9 @@ class TFRobertaForCausalLM: ...@@ -1746,6 +2226,9 @@ class TFRobertaForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRobertaForMaskedLM: class TFRobertaForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1755,6 +2238,9 @@ class TFRobertaForMaskedLM: ...@@ -1755,6 +2238,9 @@ class TFRobertaForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRobertaForMultipleChoice: class TFRobertaForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1764,6 +2250,9 @@ class TFRobertaForMultipleChoice: ...@@ -1764,6 +2250,9 @@ class TFRobertaForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRobertaForQuestionAnswering: class TFRobertaForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1773,6 +2262,9 @@ class TFRobertaForQuestionAnswering: ...@@ -1773,6 +2262,9 @@ class TFRobertaForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRobertaForSequenceClassification: class TFRobertaForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1782,6 +2274,9 @@ class TFRobertaForSequenceClassification: ...@@ -1782,6 +2274,9 @@ class TFRobertaForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRobertaForTokenClassification: class TFRobertaForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1791,6 +2286,9 @@ class TFRobertaForTokenClassification: ...@@ -1791,6 +2286,9 @@ class TFRobertaForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRobertaMainLayer: class TFRobertaMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1805,6 +2303,9 @@ class TFRobertaModel: ...@@ -1805,6 +2303,9 @@ class TFRobertaModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRobertaPreTrainedModel: class TFRobertaPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1814,6 +2315,9 @@ class TFRobertaPreTrainedModel: ...@@ -1814,6 +2315,9 @@ class TFRobertaPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_ROFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_ROFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1826,6 +2330,9 @@ class TFRoFormerForCausalLM: ...@@ -1826,6 +2330,9 @@ class TFRoFormerForCausalLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRoFormerForMaskedLM: class TFRoFormerForMaskedLM:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1835,6 +2342,9 @@ class TFRoFormerForMaskedLM: ...@@ -1835,6 +2342,9 @@ class TFRoFormerForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRoFormerForMultipleChoice: class TFRoFormerForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1844,6 +2354,9 @@ class TFRoFormerForMultipleChoice: ...@@ -1844,6 +2354,9 @@ class TFRoFormerForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRoFormerForQuestionAnswering: class TFRoFormerForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1853,6 +2366,9 @@ class TFRoFormerForQuestionAnswering: ...@@ -1853,6 +2366,9 @@ class TFRoFormerForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRoFormerForSequenceClassification: class TFRoFormerForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1862,6 +2378,9 @@ class TFRoFormerForSequenceClassification: ...@@ -1862,6 +2378,9 @@ class TFRoFormerForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRoFormerForTokenClassification: class TFRoFormerForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1871,6 +2390,9 @@ class TFRoFormerForTokenClassification: ...@@ -1871,6 +2390,9 @@ class TFRoFormerForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRoFormerLayer: class TFRoFormerLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1885,6 +2407,9 @@ class TFRoFormerModel: ...@@ -1885,6 +2407,9 @@ class TFRoFormerModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFRoFormerPreTrainedModel: class TFRoFormerPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1894,6 +2419,9 @@ class TFRoFormerPreTrainedModel: ...@@ -1894,6 +2419,9 @@ class TFRoFormerPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_T5_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_T5_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1906,6 +2434,9 @@ class TFT5EncoderModel: ...@@ -1906,6 +2434,9 @@ class TFT5EncoderModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFT5ForConditionalGeneration: class TFT5ForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1915,6 +2446,9 @@ class TFT5ForConditionalGeneration: ...@@ -1915,6 +2446,9 @@ class TFT5ForConditionalGeneration:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFT5Model: class TFT5Model:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1924,6 +2458,9 @@ class TFT5Model: ...@@ -1924,6 +2458,9 @@ class TFT5Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFT5PreTrainedModel: class TFT5PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1933,6 +2470,9 @@ class TFT5PreTrainedModel: ...@@ -1933,6 +2470,9 @@ class TFT5PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -1950,6 +2490,9 @@ class TFTransfoXLForSequenceClassification: ...@@ -1950,6 +2490,9 @@ class TFTransfoXLForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFTransfoXLLMHeadModel: class TFTransfoXLLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1959,6 +2502,9 @@ class TFTransfoXLLMHeadModel: ...@@ -1959,6 +2502,9 @@ class TFTransfoXLLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFTransfoXLMainLayer: class TFTransfoXLMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1973,6 +2519,9 @@ class TFTransfoXLModel: ...@@ -1973,6 +2519,9 @@ class TFTransfoXLModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFTransfoXLPreTrainedModel: class TFTransfoXLPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1982,6 +2531,9 @@ class TFTransfoXLPreTrainedModel: ...@@ -1982,6 +2531,9 @@ class TFTransfoXLPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFViTForImageClassification: class TFViTForImageClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -1996,6 +2548,9 @@ class TFViTModel: ...@@ -1996,6 +2548,9 @@ class TFViTModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFViTPreTrainedModel: class TFViTPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2005,6 +2560,9 @@ class TFViTPreTrainedModel: ...@@ -2005,6 +2560,9 @@ class TFViTPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2022,6 +2580,9 @@ class TFWav2Vec2Model: ...@@ -2022,6 +2580,9 @@ class TFWav2Vec2Model:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFWav2Vec2PreTrainedModel: class TFWav2Vec2PreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2031,6 +2592,9 @@ class TFWav2Vec2PreTrainedModel: ...@@ -2031,6 +2592,9 @@ class TFWav2Vec2PreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2043,6 +2607,9 @@ class TFXLMForMultipleChoice: ...@@ -2043,6 +2607,9 @@ class TFXLMForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMForQuestionAnsweringSimple: class TFXLMForQuestionAnsweringSimple:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2052,6 +2619,9 @@ class TFXLMForQuestionAnsweringSimple: ...@@ -2052,6 +2619,9 @@ class TFXLMForQuestionAnsweringSimple:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMForSequenceClassification: class TFXLMForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2061,6 +2631,9 @@ class TFXLMForSequenceClassification: ...@@ -2061,6 +2631,9 @@ class TFXLMForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMForTokenClassification: class TFXLMForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2070,6 +2643,9 @@ class TFXLMForTokenClassification: ...@@ -2070,6 +2643,9 @@ class TFXLMForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMMainLayer: class TFXLMMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2084,6 +2660,9 @@ class TFXLMModel: ...@@ -2084,6 +2660,9 @@ class TFXLMModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMPreTrainedModel: class TFXLMPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2093,6 +2672,9 @@ class TFXLMPreTrainedModel: ...@@ -2093,6 +2672,9 @@ class TFXLMPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMWithLMHeadModel: class TFXLMWithLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2102,6 +2684,9 @@ class TFXLMWithLMHeadModel: ...@@ -2102,6 +2684,9 @@ class TFXLMWithLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2114,6 +2699,9 @@ class TFXLMRobertaForMaskedLM: ...@@ -2114,6 +2699,9 @@ class TFXLMRobertaForMaskedLM:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMRobertaForMultipleChoice: class TFXLMRobertaForMultipleChoice:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2123,6 +2711,9 @@ class TFXLMRobertaForMultipleChoice: ...@@ -2123,6 +2711,9 @@ class TFXLMRobertaForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMRobertaForQuestionAnswering: class TFXLMRobertaForQuestionAnswering:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2132,6 +2723,9 @@ class TFXLMRobertaForQuestionAnswering: ...@@ -2132,6 +2723,9 @@ class TFXLMRobertaForQuestionAnswering:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMRobertaForSequenceClassification: class TFXLMRobertaForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2141,6 +2735,9 @@ class TFXLMRobertaForSequenceClassification: ...@@ -2141,6 +2735,9 @@ class TFXLMRobertaForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMRobertaForTokenClassification: class TFXLMRobertaForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2150,6 +2747,9 @@ class TFXLMRobertaForTokenClassification: ...@@ -2150,6 +2747,9 @@ class TFXLMRobertaForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLMRobertaModel: class TFXLMRobertaModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2159,6 +2759,9 @@ class TFXLMRobertaModel: ...@@ -2159,6 +2759,9 @@ class TFXLMRobertaModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_XLNET_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_XLNET_PRETRAINED_MODEL_ARCHIVE_LIST = None
...@@ -2171,6 +2774,9 @@ class TFXLNetForMultipleChoice: ...@@ -2171,6 +2774,9 @@ class TFXLNetForMultipleChoice:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLNetForQuestionAnsweringSimple: class TFXLNetForQuestionAnsweringSimple:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2180,6 +2786,9 @@ class TFXLNetForQuestionAnsweringSimple: ...@@ -2180,6 +2786,9 @@ class TFXLNetForQuestionAnsweringSimple:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLNetForSequenceClassification: class TFXLNetForSequenceClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2189,6 +2798,9 @@ class TFXLNetForSequenceClassification: ...@@ -2189,6 +2798,9 @@ class TFXLNetForSequenceClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLNetForTokenClassification: class TFXLNetForTokenClassification:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2198,6 +2810,9 @@ class TFXLNetForTokenClassification: ...@@ -2198,6 +2810,9 @@ class TFXLNetForTokenClassification:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLNetLMHeadModel: class TFXLNetLMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2207,6 +2822,9 @@ class TFXLNetLMHeadModel: ...@@ -2207,6 +2822,9 @@ class TFXLNetLMHeadModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLNetMainLayer: class TFXLNetMainLayer:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2221,6 +2839,9 @@ class TFXLNetModel: ...@@ -2221,6 +2839,9 @@ class TFXLNetModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFXLNetPreTrainedModel: class TFXLNetPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -2230,6 +2851,9 @@ class TFXLNetPreTrainedModel: ...@@ -2230,6 +2851,9 @@ class TFXLNetPreTrainedModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["tf"]) requires_backends(cls, ["tf"])
def call(self, *args, **kwargs):
requires_backends(self, ["tf"])
class AdamWeightDecay: class AdamWeightDecay:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
......
src/transformers/utils/dummy_timm_and_vision_objects.py
View file @ 3e8d17e6
...@@ -13,6 +13,9 @@ class DetrForObjectDetection: ...@@ -13,6 +13,9 @@ class DetrForObjectDetection:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["timm", "vision"]) requires_backends(cls, ["timm", "vision"])
def forward(self, *args, **kwargs):
requires_backends(self, ["timm", "vision"])
class DetrForSegmentation: class DetrForSegmentation:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -22,6 +25,9 @@ class DetrForSegmentation: ...@@ -22,6 +25,9 @@ class DetrForSegmentation:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["timm", "vision"]) requires_backends(cls, ["timm", "vision"])
def forward(self, *args, **kwargs):
requires_backends(self, ["timm", "vision"])
class DetrModel: class DetrModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -31,6 +37,9 @@ class DetrModel: ...@@ -31,6 +37,9 @@ class DetrModel:
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["timm", "vision"]) requires_backends(cls, ["timm", "vision"])
def forward(self, *args, **kwargs):
requires_backends(self, ["timm", "vision"])
class DetrPreTrainedModel: class DetrPreTrainedModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -39,3 +48,6 @@ class DetrPreTrainedModel: ...@@ -39,3 +48,6 @@ class DetrPreTrainedModel:
@classmethod @classmethod
def from_pretrained(cls, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["timm", "vision"]) requires_backends(cls, ["timm", "vision"])
def forward(self, *args, **kwargs):
requires_backends(self, ["timm", "vision"])
utils/check_dummies.py
View file @ 3e8d17e6
...@@ -43,6 +43,30 @@ class {0}: ...@@ -43,6 +43,30 @@ class {0}:
requires_backends(cls, {1}) requires_backends(cls, {1})
""" """
PT_DUMMY_PRETRAINED_CLASS = (
DUMMY_PRETRAINED_CLASS
+ """
def forward(self, *args, **kwargs):
requires_backends(self, {1})
"""
)
TF_DUMMY_PRETRAINED_CLASS = (
DUMMY_PRETRAINED_CLASS
+ """
def call(self, *args, **kwargs):
requires_backends(self, {1})
"""
)
FLAX_DUMMY_PRETRAINED_CLASS = (
DUMMY_PRETRAINED_CLASS
+ """
def __call__(self, *args, **kwargs):
requires_backends(self, {1})
"""
)
DUMMY_CLASS = """ DUMMY_CLASS = """
class {0}: class {0}:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
...@@ -102,8 +126,7 @@ def read_init(): ...@@ -102,8 +126,7 @@ def read_init():
def create_dummy_object(name, backend_name): def create_dummy_object(name, backend_name):
"""Create the code for the dummy object corresponding to `name`.""" """Create the code for the dummy object corresponding to `name`."""
_pretrained = [ _models = [
"Config",
"ForCausalLM", "ForCausalLM",
"ForConditionalGeneration", "ForConditionalGeneration",
"ForMaskedLM", "ForMaskedLM",
...@@ -114,14 +137,24 @@ def create_dummy_object(name, backend_name): ...@@ -114,14 +137,24 @@ def create_dummy_object(name, backend_name):
"ForSequenceClassification", "ForSequenceClassification",
"ForTokenClassification", "ForTokenClassification",
"Model", "Model",
"Tokenizer",
"Processor",
] ]
_pretrained = ["Config", "Tokenizer", "Processor"]
if name.isupper(): if name.isupper():
return DUMMY_CONSTANT.format(name) return DUMMY_CONSTANT.format(name)
elif name.islower(): elif name.islower():
return DUMMY_FUNCTION.format(name, backend_name) return DUMMY_FUNCTION.format(name, backend_name)
else: else:
is_model = False
for part in _models:
if part in name:
is_model = True
break
if is_model:
if name.startswith("TF"):
return TF_DUMMY_PRETRAINED_CLASS.format(name, backend_name)
if name.startswith("Flax"):
return FLAX_DUMMY_PRETRAINED_CLASS.format(name, backend_name)
return PT_DUMMY_PRETRAINED_CLASS.format(name, backend_name)
is_pretrained = False is_pretrained = False
for part in _pretrained: for part in _pretrained:
if part in name: if part in name:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment