mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
Add forward method to dummy models (#14419)
* Add forward method to dummy models * Fix quality
This commit is contained in:
Sylvain Gugger
GitHub
parent
040fd47162
commit
3e8d17e66d
@ -75,6 +75,9 @@ class FlaxPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAlbertForMaskedLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -84,6 +87,9 @@ class FlaxAlbertForMaskedLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAlbertForMultipleChoice:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -93,6 +99,9 @@ class FlaxAlbertForMultipleChoice:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAlbertForPreTraining:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -107,6 +116,9 @@ class FlaxAlbertForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAlbertForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -116,6 +128,9 @@ class FlaxAlbertForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAlbertForTokenClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -125,6 +140,9 @@ class FlaxAlbertForTokenClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAlbertModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -134,6 +152,9 @@ class FlaxAlbertModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAlbertPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -143,6 +164,9 @@ class FlaxAlbertPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
FLAX_MODEL_FOR_CAUSAL_LM_MAPPING = None
|
||||
|
||||
@ -188,6 +212,9 @@ class FlaxAutoModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForCausalLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -197,6 +224,9 @@ class FlaxAutoModelForCausalLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForImageClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -206,6 +236,9 @@ class FlaxAutoModelForImageClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForMaskedLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -215,6 +248,9 @@ class FlaxAutoModelForMaskedLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForMultipleChoice:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -224,6 +260,9 @@ class FlaxAutoModelForMultipleChoice:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForNextSentencePrediction:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -233,6 +272,9 @@ class FlaxAutoModelForNextSentencePrediction:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForPreTraining:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -242,6 +284,9 @@ class FlaxAutoModelForPreTraining:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForQuestionAnswering:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -251,6 +296,9 @@ class FlaxAutoModelForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForSeq2SeqLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -260,6 +308,9 @@ class FlaxAutoModelForSeq2SeqLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -269,6 +320,9 @@ class FlaxAutoModelForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForTokenClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -278,6 +332,9 @@ class FlaxAutoModelForTokenClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxAutoModelForVision2Seq:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -287,6 +344,9 @@ class FlaxAutoModelForVision2Seq:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBartForConditionalGeneration:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -296,6 +356,9 @@ class FlaxBartForConditionalGeneration:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBartForQuestionAnswering:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -305,6 +368,9 @@ class FlaxBartForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBartForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -314,6 +380,9 @@ class FlaxBartForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBartModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -323,6 +392,9 @@ class FlaxBartModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBartPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -332,6 +404,9 @@ class FlaxBartPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBeitForImageClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -346,6 +421,9 @@ class FlaxBeitForMaskedImageModeling:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBeitModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -355,6 +433,9 @@ class FlaxBeitModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBeitPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -364,6 +445,9 @@ class FlaxBeitPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBertForMaskedLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -373,6 +457,9 @@ class FlaxBertForMaskedLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBertForMultipleChoice:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -382,6 +469,9 @@ class FlaxBertForMultipleChoice:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBertForNextSentencePrediction:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -401,6 +491,9 @@ class FlaxBertForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBertForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -410,6 +503,9 @@ class FlaxBertForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBertForTokenClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -419,6 +515,9 @@ class FlaxBertForTokenClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBertModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -428,6 +527,9 @@ class FlaxBertModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBertPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -437,6 +539,9 @@ class FlaxBertPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBigBirdForMaskedLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -446,6 +551,9 @@ class FlaxBigBirdForMaskedLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBigBirdForMultipleChoice:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -455,6 +563,9 @@ class FlaxBigBirdForMultipleChoice:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBigBirdForPreTraining:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -469,6 +580,9 @@ class FlaxBigBirdForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBigBirdForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -478,6 +592,9 @@ class FlaxBigBirdForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBigBirdForTokenClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -487,6 +604,9 @@ class FlaxBigBirdForTokenClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBigBirdModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -496,6 +616,9 @@ class FlaxBigBirdModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxBigBirdPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -505,6 +628,9 @@ class FlaxBigBirdPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxCLIPModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -514,6 +640,9 @@ class FlaxCLIPModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxCLIPPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -523,6 +652,9 @@ class FlaxCLIPPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxCLIPTextModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -532,6 +664,9 @@ class FlaxCLIPTextModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxCLIPTextPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -541,6 +676,9 @@ class FlaxCLIPTextPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxCLIPVisionModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -550,6 +688,9 @@ class FlaxCLIPVisionModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxCLIPVisionPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -559,6 +700,9 @@ class FlaxCLIPVisionPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxDistilBertForMaskedLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -568,6 +712,9 @@ class FlaxDistilBertForMaskedLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxDistilBertForMultipleChoice:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -577,6 +724,9 @@ class FlaxDistilBertForMultipleChoice:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxDistilBertForQuestionAnswering:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -586,6 +736,9 @@ class FlaxDistilBertForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxDistilBertForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -595,6 +748,9 @@ class FlaxDistilBertForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxDistilBertForTokenClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -604,6 +760,9 @@ class FlaxDistilBertForTokenClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxDistilBertModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -613,6 +772,9 @@ class FlaxDistilBertModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxDistilBertPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -622,6 +784,9 @@ class FlaxDistilBertPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxElectraForMaskedLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -631,6 +796,9 @@ class FlaxElectraForMaskedLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxElectraForMultipleChoice:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -640,6 +808,9 @@ class FlaxElectraForMultipleChoice:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxElectraForPreTraining:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -654,6 +825,9 @@ class FlaxElectraForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxElectraForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -663,6 +837,9 @@ class FlaxElectraForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxElectraForTokenClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -672,6 +849,9 @@ class FlaxElectraForTokenClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxElectraModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -681,6 +861,9 @@ class FlaxElectraModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxElectraPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -690,6 +873,9 @@ class FlaxElectraPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxEncoderDecoderModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -699,6 +885,9 @@ class FlaxEncoderDecoderModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxGPT2LMHeadModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -708,6 +897,9 @@ class FlaxGPT2LMHeadModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxGPT2Model:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -717,6 +909,9 @@ class FlaxGPT2Model:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxGPT2PreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -726,6 +921,9 @@ class FlaxGPT2PreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxGPTNeoForCausalLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -735,6 +933,9 @@ class FlaxGPTNeoForCausalLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxGPTNeoModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -744,6 +945,9 @@ class FlaxGPTNeoModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxGPTNeoPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -753,6 +957,9 @@ class FlaxGPTNeoPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMarianModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -762,6 +969,9 @@ class FlaxMarianModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMarianMTModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -771,6 +981,9 @@ class FlaxMarianMTModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMarianPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -780,6 +993,9 @@ class FlaxMarianPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMBartForConditionalGeneration:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -789,6 +1005,9 @@ class FlaxMBartForConditionalGeneration:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMBartForQuestionAnswering:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -798,6 +1017,9 @@ class FlaxMBartForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMBartForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -807,6 +1029,9 @@ class FlaxMBartForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMBartModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -816,6 +1041,9 @@ class FlaxMBartModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMBartPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -825,6 +1053,9 @@ class FlaxMBartPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMT5ForConditionalGeneration:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -834,6 +1065,9 @@ class FlaxMT5ForConditionalGeneration:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxMT5Model:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -843,6 +1077,9 @@ class FlaxMT5Model:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxPegasusForConditionalGeneration:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -852,6 +1089,9 @@ class FlaxPegasusForConditionalGeneration:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxPegasusModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -861,6 +1101,9 @@ class FlaxPegasusModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxPegasusPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -870,6 +1113,9 @@ class FlaxPegasusPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxRobertaForMaskedLM:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -879,6 +1125,9 @@ class FlaxRobertaForMaskedLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxRobertaForMultipleChoice:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -888,6 +1137,9 @@ class FlaxRobertaForMultipleChoice:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxRobertaForQuestionAnswering:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -897,6 +1149,9 @@ class FlaxRobertaForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxRobertaForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -906,6 +1161,9 @@ class FlaxRobertaForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxRobertaForTokenClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -915,6 +1173,9 @@ class FlaxRobertaForTokenClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxRobertaModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -924,6 +1185,9 @@ class FlaxRobertaModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxRobertaPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -933,6 +1197,9 @@ class FlaxRobertaPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxT5ForConditionalGeneration:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -942,6 +1209,9 @@ class FlaxT5ForConditionalGeneration:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxT5Model:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -951,6 +1221,9 @@ class FlaxT5Model:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxT5PreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -960,6 +1233,9 @@ class FlaxT5PreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxVisionEncoderDecoderModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -969,6 +1245,9 @@ class FlaxVisionEncoderDecoderModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxViTForImageClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -983,6 +1262,9 @@ class FlaxViTModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxViTPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -992,6 +1274,9 @@ class FlaxViTPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxWav2Vec2ForCTC:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -1011,6 +1296,9 @@ class FlaxWav2Vec2Model:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
|
||||
class FlaxWav2Vec2PreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -1019,3 +1307,6 @@ class FlaxWav2Vec2PreTrainedModel:
|
||||
@classmethod
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["flax"])
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
requires_backends(self, ["flax"])
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@ -13,6 +13,9 @@ class TapasForMaskedLM:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["scatter"])
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
requires_backends(self, ["scatter"])
|
||||
|
||||
|
||||
class TapasForQuestionAnswering:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -22,6 +25,9 @@ class TapasForQuestionAnswering:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["scatter"])
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
requires_backends(self, ["scatter"])
|
||||
|
||||
|
||||
class TapasForSequenceClassification:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -31,6 +37,9 @@ class TapasForSequenceClassification:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["scatter"])
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
requires_backends(self, ["scatter"])
|
||||
|
||||
|
||||
class TapasModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -40,6 +49,9 @@ class TapasModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["scatter"])
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
requires_backends(self, ["scatter"])
|
||||
|
||||
|
||||
class TapasPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -49,6 +61,9 @@ class TapasPreTrainedModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["scatter"])
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
requires_backends(self, ["scatter"])
|
||||
|
||||
|
||||
def load_tf_weights_in_tapas(*args, **kwargs):
|
||||
requires_backends(load_tf_weights_in_tapas, ["scatter"])
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@ -13,6 +13,9 @@ class DetrForObjectDetection:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["timm", "vision"])
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
requires_backends(self, ["timm", "vision"])
|
||||
|
||||
|
||||
class DetrForSegmentation:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -22,6 +25,9 @@ class DetrForSegmentation:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["timm", "vision"])
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
requires_backends(self, ["timm", "vision"])
|
||||
|
||||
|
||||
class DetrModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -31,6 +37,9 @@ class DetrModel:
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["timm", "vision"])
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
requires_backends(self, ["timm", "vision"])
|
||||
|
||||
|
||||
class DetrPreTrainedModel:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -39,3 +48,6 @@ class DetrPreTrainedModel:
|
||||
@classmethod
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["timm", "vision"])
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
requires_backends(self, ["timm", "vision"])
|
||||
|
||||
@ -43,6 +43,30 @@ class {0}:
|
||||
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 = """
|
||||
class {0}:
|
||||
def __init__(self, *args, **kwargs):
|
||||
@ -102,8 +126,7 @@ def read_init():
|
||||
|
||||
def create_dummy_object(name, backend_name):
|
||||
"""Create the code for the dummy object corresponding to `name`."""
|
||||
_pretrained = [
|
||||
"Config",
|
||||
_models = [
|
||||
"ForCausalLM",
|
||||
"ForConditionalGeneration",
|
||||
"ForMaskedLM",
|
||||
@ -114,14 +137,24 @@ def create_dummy_object(name, backend_name):
|
||||
"ForSequenceClassification",
|
||||
"ForTokenClassification",
|
||||
"Model",
|
||||
"Tokenizer",
|
||||
"Processor",
|
||||
]
|
||||
_pretrained = ["Config", "Tokenizer", "Processor"]
|
||||
if name.isupper():
|
||||
return DUMMY_CONSTANT.format(name)
|
||||
elif name.islower():
|
||||
return DUMMY_FUNCTION.format(name, backend_name)
|
||||
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
|
||||
for part in _pretrained:
|
||||
if part in name:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user