SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add forward method to dummy models (#14419)

Browse Source 
* Add forward method to dummy models

* Fix quality
This commit is contained in:
Sylvain Gugger
2021-11-16 09:24:40 -05:00
committed by GitHub
parent 040fd47162
commit 3e8d17e66d
6 changed files with 2071 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

291
src/transformers/utils/dummy_flax_objects.py
View File

@@ -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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
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:
@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"])

1092
src/transformers/utils/dummy_pt_objects.py
View File

File diff suppressed because it is too large Load Diff

15
src/transformers/utils/dummy_scatter_objects.py
View File

@@ -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:
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:
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:
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:
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"])

624
src/transformers/utils/dummy_tf_objects.py
View File

File diff suppressed because it is too large Load Diff

12
src/transformers/utils/dummy_timm_and_vision_objects.py
View File

@@ -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:
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:
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:
@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"])

41
utils/check_dummies.py
View File

@@ -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():
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):
"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: