Skip to content

GitLab

Unverified Commit 3b1f5caf authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Add from_pretrained to dummy timm objects (#12097) 



* Add from_pretrained to dummy timm

* Fix at the source

* Update utils/check_dummies.py
Co-authored-by: default avatarLysandre Debut <lysandre@huggingface.co>

* Missing pretrained dummies

* Style
Co-authored-by: default avatarSylvain Gugger <sylvain.gugger@gmail.com>
Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
parent 15b498f3
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 1158 additions and 1071 deletions
src/transformers/utils/dummy_flax_objects.py
View file @ 3b1f5caf
...@@ -37,8 +37,8 @@ class FlaxPreTrainedModel: ...@@ -37,8 +37,8 @@ class FlaxPreTrainedModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
FLAX_MODEL_FOR_CAUSAL_LM_MAPPING = None FLAX_MODEL_FOR_CAUSAL_LM_MAPPING = None
...@@ -73,8 +73,8 @@ class FlaxAutoModel: ...@@ -73,8 +73,8 @@ class FlaxAutoModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxAutoModelForCausalLM: class FlaxAutoModelForCausalLM:
...@@ -82,8 +82,8 @@ class FlaxAutoModelForCausalLM: ...@@ -82,8 +82,8 @@ class FlaxAutoModelForCausalLM:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxAutoModelForMaskedLM: class FlaxAutoModelForMaskedLM:
...@@ -91,8 +91,8 @@ class FlaxAutoModelForMaskedLM: ...@@ -91,8 +91,8 @@ class FlaxAutoModelForMaskedLM:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxAutoModelForMultipleChoice: class FlaxAutoModelForMultipleChoice:
...@@ -100,8 +100,8 @@ class FlaxAutoModelForMultipleChoice: ...@@ -100,8 +100,8 @@ class FlaxAutoModelForMultipleChoice:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxAutoModelForNextSentencePrediction: class FlaxAutoModelForNextSentencePrediction:
...@@ -109,8 +109,8 @@ class FlaxAutoModelForNextSentencePrediction: ...@@ -109,8 +109,8 @@ class FlaxAutoModelForNextSentencePrediction:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxAutoModelForPreTraining: class FlaxAutoModelForPreTraining:
...@@ -118,8 +118,8 @@ class FlaxAutoModelForPreTraining: ...@@ -118,8 +118,8 @@ class FlaxAutoModelForPreTraining:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxAutoModelForQuestionAnswering: class FlaxAutoModelForQuestionAnswering:
...@@ -127,8 +127,8 @@ class FlaxAutoModelForQuestionAnswering: ...@@ -127,8 +127,8 @@ class FlaxAutoModelForQuestionAnswering:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxAutoModelForSequenceClassification: class FlaxAutoModelForSequenceClassification:
...@@ -136,8 +136,8 @@ class FlaxAutoModelForSequenceClassification: ...@@ -136,8 +136,8 @@ class FlaxAutoModelForSequenceClassification:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxAutoModelForTokenClassification: class FlaxAutoModelForTokenClassification:
...@@ -145,8 +145,8 @@ class FlaxAutoModelForTokenClassification: ...@@ -145,8 +145,8 @@ class FlaxAutoModelForTokenClassification:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxBertForMaskedLM: class FlaxBertForMaskedLM:
...@@ -154,8 +154,8 @@ class FlaxBertForMaskedLM: ...@@ -154,8 +154,8 @@ class FlaxBertForMaskedLM:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxBertForMultipleChoice: class FlaxBertForMultipleChoice:
...@@ -163,8 +163,8 @@ class FlaxBertForMultipleChoice: ...@@ -163,8 +163,8 @@ class FlaxBertForMultipleChoice:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxBertForNextSentencePrediction: class FlaxBertForNextSentencePrediction:
...@@ -182,8 +182,8 @@ class FlaxBertForQuestionAnswering: ...@@ -182,8 +182,8 @@ class FlaxBertForQuestionAnswering:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxBertForSequenceClassification: class FlaxBertForSequenceClassification:
...@@ -191,8 +191,8 @@ class FlaxBertForSequenceClassification: ...@@ -191,8 +191,8 @@ class FlaxBertForSequenceClassification:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxBertForTokenClassification: class FlaxBertForTokenClassification:
...@@ -200,8 +200,8 @@ class FlaxBertForTokenClassification: ...@@ -200,8 +200,8 @@ class FlaxBertForTokenClassification:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxBertModel: class FlaxBertModel:
...@@ -209,8 +209,8 @@ class FlaxBertModel: ...@@ -209,8 +209,8 @@ class FlaxBertModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxBertPreTrainedModel: class FlaxBertPreTrainedModel:
...@@ -218,8 +218,8 @@ class FlaxBertPreTrainedModel: ...@@ -218,8 +218,8 @@ class FlaxBertPreTrainedModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxCLIPModel: class FlaxCLIPModel:
...@@ -227,8 +227,8 @@ class FlaxCLIPModel: ...@@ -227,8 +227,8 @@ class FlaxCLIPModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxCLIPPreTrainedModel: class FlaxCLIPPreTrainedModel:
...@@ -236,8 +236,8 @@ class FlaxCLIPPreTrainedModel: ...@@ -236,8 +236,8 @@ class FlaxCLIPPreTrainedModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxCLIPTextModel: class FlaxCLIPTextModel:
...@@ -245,8 +245,8 @@ class FlaxCLIPTextModel: ...@@ -245,8 +245,8 @@ class FlaxCLIPTextModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxCLIPVisionModel: class FlaxCLIPVisionModel:
...@@ -254,8 +254,8 @@ class FlaxCLIPVisionModel: ...@@ -254,8 +254,8 @@ class FlaxCLIPVisionModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxElectraForMaskedLM: class FlaxElectraForMaskedLM:
...@@ -263,8 +263,8 @@ class FlaxElectraForMaskedLM: ...@@ -263,8 +263,8 @@ class FlaxElectraForMaskedLM:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxElectraForMultipleChoice: class FlaxElectraForMultipleChoice:
...@@ -272,8 +272,8 @@ class FlaxElectraForMultipleChoice: ...@@ -272,8 +272,8 @@ class FlaxElectraForMultipleChoice:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxElectraForPreTraining: class FlaxElectraForPreTraining:
...@@ -286,8 +286,8 @@ class FlaxElectraForQuestionAnswering: ...@@ -286,8 +286,8 @@ class FlaxElectraForQuestionAnswering:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxElectraForSequenceClassification: class FlaxElectraForSequenceClassification:
...@@ -295,8 +295,8 @@ class FlaxElectraForSequenceClassification: ...@@ -295,8 +295,8 @@ class FlaxElectraForSequenceClassification:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxElectraForTokenClassification: class FlaxElectraForTokenClassification:
...@@ -304,8 +304,8 @@ class FlaxElectraForTokenClassification: ...@@ -304,8 +304,8 @@ class FlaxElectraForTokenClassification:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxElectraModel: class FlaxElectraModel:
...@@ -313,8 +313,8 @@ class FlaxElectraModel: ...@@ -313,8 +313,8 @@ class FlaxElectraModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxElectraPreTrainedModel: class FlaxElectraPreTrainedModel:
...@@ -322,8 +322,8 @@ class FlaxElectraPreTrainedModel: ...@@ -322,8 +322,8 @@ class FlaxElectraPreTrainedModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxGPT2LMHeadModel: class FlaxGPT2LMHeadModel:
...@@ -331,8 +331,8 @@ class FlaxGPT2LMHeadModel: ...@@ -331,8 +331,8 @@ class FlaxGPT2LMHeadModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxGPT2Model: class FlaxGPT2Model:
...@@ -340,8 +340,8 @@ class FlaxGPT2Model: ...@@ -340,8 +340,8 @@ class FlaxGPT2Model:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxRobertaForMaskedLM: class FlaxRobertaForMaskedLM:
...@@ -349,8 +349,8 @@ class FlaxRobertaForMaskedLM: ...@@ -349,8 +349,8 @@ class FlaxRobertaForMaskedLM:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxRobertaForMultipleChoice: class FlaxRobertaForMultipleChoice:
...@@ -358,8 +358,8 @@ class FlaxRobertaForMultipleChoice: ...@@ -358,8 +358,8 @@ class FlaxRobertaForMultipleChoice:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxRobertaForQuestionAnswering: class FlaxRobertaForQuestionAnswering:
...@@ -367,8 +367,8 @@ class FlaxRobertaForQuestionAnswering: ...@@ -367,8 +367,8 @@ class FlaxRobertaForQuestionAnswering:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxRobertaForSequenceClassification: class FlaxRobertaForSequenceClassification:
...@@ -376,8 +376,8 @@ class FlaxRobertaForSequenceClassification: ...@@ -376,8 +376,8 @@ class FlaxRobertaForSequenceClassification:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxRobertaForTokenClassification: class FlaxRobertaForTokenClassification:
...@@ -385,8 +385,8 @@ class FlaxRobertaForTokenClassification: ...@@ -385,8 +385,8 @@ class FlaxRobertaForTokenClassification:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxRobertaModel: class FlaxRobertaModel:
...@@ -394,8 +394,8 @@ class FlaxRobertaModel: ...@@ -394,8 +394,8 @@ class FlaxRobertaModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxRobertaPreTrainedModel: class FlaxRobertaPreTrainedModel:
...@@ -403,8 +403,8 @@ class FlaxRobertaPreTrainedModel: ...@@ -403,8 +403,8 @@ class FlaxRobertaPreTrainedModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
class FlaxViTForImageClassification: class FlaxViTForImageClassification:
...@@ -417,5 +417,5 @@ class FlaxViTModel: ...@@ -417,5 +417,5 @@ class FlaxViTModel:
requires_backends(self, ["flax"]) requires_backends(self, ["flax"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["flax"]) requires_backends(cls, ["flax"])
src/transformers/utils/dummy_pt_objects.py
View file @ 3b1f5caf
This diff is collapsed.
src/transformers/utils/dummy_sentencepiece_objects.py
View file @ 3b1f5caf
...@@ -7,8 +7,8 @@ class AlbertTokenizer: ...@@ -7,8 +7,8 @@ class AlbertTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class BarthezTokenizer: class BarthezTokenizer:
...@@ -16,8 +16,8 @@ class BarthezTokenizer: ...@@ -16,8 +16,8 @@ class BarthezTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class BertGenerationTokenizer: class BertGenerationTokenizer:
...@@ -25,8 +25,8 @@ class BertGenerationTokenizer: ...@@ -25,8 +25,8 @@ class BertGenerationTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class CamembertTokenizer: class CamembertTokenizer:
...@@ -34,8 +34,8 @@ class CamembertTokenizer: ...@@ -34,8 +34,8 @@ class CamembertTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class DebertaV2Tokenizer: class DebertaV2Tokenizer:
...@@ -43,8 +43,8 @@ class DebertaV2Tokenizer: ...@@ -43,8 +43,8 @@ class DebertaV2Tokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class M2M100Tokenizer: class M2M100Tokenizer:
...@@ -52,8 +52,8 @@ class M2M100Tokenizer: ...@@ -52,8 +52,8 @@ class M2M100Tokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class MarianTokenizer: class MarianTokenizer:
...@@ -61,8 +61,8 @@ class MarianTokenizer: ...@@ -61,8 +61,8 @@ class MarianTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class MBart50Tokenizer: class MBart50Tokenizer:
...@@ -70,8 +70,8 @@ class MBart50Tokenizer: ...@@ -70,8 +70,8 @@ class MBart50Tokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class MBartTokenizer: class MBartTokenizer:
...@@ -79,8 +79,8 @@ class MBartTokenizer: ...@@ -79,8 +79,8 @@ class MBartTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class MT5Tokenizer: class MT5Tokenizer:
...@@ -88,8 +88,8 @@ class MT5Tokenizer: ...@@ -88,8 +88,8 @@ class MT5Tokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class PegasusTokenizer: class PegasusTokenizer:
...@@ -97,8 +97,8 @@ class PegasusTokenizer: ...@@ -97,8 +97,8 @@ class PegasusTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class ReformerTokenizer: class ReformerTokenizer:
...@@ -106,8 +106,8 @@ class ReformerTokenizer: ...@@ -106,8 +106,8 @@ class ReformerTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class Speech2TextTokenizer: class Speech2TextTokenizer:
...@@ -115,8 +115,8 @@ class Speech2TextTokenizer: ...@@ -115,8 +115,8 @@ class Speech2TextTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class T5Tokenizer: class T5Tokenizer:
...@@ -124,8 +124,8 @@ class T5Tokenizer: ...@@ -124,8 +124,8 @@ class T5Tokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class XLMProphetNetTokenizer: class XLMProphetNetTokenizer:
...@@ -133,8 +133,8 @@ class XLMProphetNetTokenizer: ...@@ -133,8 +133,8 @@ class XLMProphetNetTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class XLMRobertaTokenizer: class XLMRobertaTokenizer:
...@@ -142,8 +142,8 @@ class XLMRobertaTokenizer: ...@@ -142,8 +142,8 @@ class XLMRobertaTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
class XLNetTokenizer: class XLNetTokenizer:
...@@ -151,5 +151,5 @@ class XLNetTokenizer: ...@@ -151,5 +151,5 @@ class XLNetTokenizer:
requires_backends(self, ["sentencepiece"]) requires_backends(self, ["sentencepiece"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["sentencepiece"]) requires_backends(cls, ["sentencepiece"])
src/transformers/utils/dummy_tf_objects.py
View file @ 3b1f5caf
This diff is collapsed.
src/transformers/utils/dummy_timm_and_vision_objects.py
View file @ 3b1f5caf
...@@ -9,16 +9,24 @@ class DetrForObjectDetection: ...@@ -9,16 +9,24 @@ class DetrForObjectDetection:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
requires_backends(self, ["timm", "vision"]) requires_backends(self, ["timm", "vision"])
@classmethod
def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["timm", "vision"])
class DetrForSegmentation: class DetrForSegmentation:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
requires_backends(self, ["timm", "vision"]) requires_backends(self, ["timm", "vision"])
@classmethod
def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["timm", "vision"])
class DetrModel: class DetrModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
requires_backends(self, ["timm", "vision"]) requires_backends(self, ["timm", "vision"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["timm", "vision"]) requires_backends(cls, ["timm", "vision"])
src/transformers/utils/dummy_timm_objects.py
View file @ 3b1f5caf
...@@ -9,16 +9,24 @@ class DetrForObjectDetection: ...@@ -9,16 +9,24 @@ class DetrForObjectDetection:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
requires_backends(self, ["timm"]) requires_backends(self, ["timm"])
@classmethod
def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["timm"])
class DetrForSegmentation: class DetrForSegmentation:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
requires_backends(self, ["timm"]) requires_backends(self, ["timm"])
@classmethod
def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["timm"])
class DetrModel: class DetrModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
requires_backends(self, ["timm"]) requires_backends(self, ["timm"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["timm"]) requires_backends(cls, ["timm"])
src/transformers/utils/dummy_tokenizers_objects.py
View file @ 3b1f5caf
...@@ -7,8 +7,8 @@ class AlbertTokenizerFast: ...@@ -7,8 +7,8 @@ class AlbertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class BartTokenizerFast: class BartTokenizerFast:
...@@ -16,8 +16,8 @@ class BartTokenizerFast: ...@@ -16,8 +16,8 @@ class BartTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class BarthezTokenizerFast: class BarthezTokenizerFast:
...@@ -25,8 +25,8 @@ class BarthezTokenizerFast: ...@@ -25,8 +25,8 @@ class BarthezTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class BertTokenizerFast: class BertTokenizerFast:
...@@ -34,8 +34,8 @@ class BertTokenizerFast: ...@@ -34,8 +34,8 @@ class BertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class BigBirdTokenizerFast: class BigBirdTokenizerFast:
...@@ -43,8 +43,8 @@ class BigBirdTokenizerFast: ...@@ -43,8 +43,8 @@ class BigBirdTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class CamembertTokenizerFast: class CamembertTokenizerFast:
...@@ -52,8 +52,8 @@ class CamembertTokenizerFast: ...@@ -52,8 +52,8 @@ class CamembertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class CLIPTokenizerFast: class CLIPTokenizerFast:
...@@ -61,8 +61,8 @@ class CLIPTokenizerFast: ...@@ -61,8 +61,8 @@ class CLIPTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class ConvBertTokenizerFast: class ConvBertTokenizerFast:
...@@ -70,8 +70,8 @@ class ConvBertTokenizerFast: ...@@ -70,8 +70,8 @@ class ConvBertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class DebertaTokenizerFast: class DebertaTokenizerFast:
...@@ -79,8 +79,8 @@ class DebertaTokenizerFast: ...@@ -79,8 +79,8 @@ class DebertaTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class DistilBertTokenizerFast: class DistilBertTokenizerFast:
...@@ -88,8 +88,8 @@ class DistilBertTokenizerFast: ...@@ -88,8 +88,8 @@ class DistilBertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class DPRContextEncoderTokenizerFast: class DPRContextEncoderTokenizerFast:
...@@ -97,8 +97,8 @@ class DPRContextEncoderTokenizerFast: ...@@ -97,8 +97,8 @@ class DPRContextEncoderTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class DPRQuestionEncoderTokenizerFast: class DPRQuestionEncoderTokenizerFast:
...@@ -106,8 +106,8 @@ class DPRQuestionEncoderTokenizerFast: ...@@ -106,8 +106,8 @@ class DPRQuestionEncoderTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class DPRReaderTokenizerFast: class DPRReaderTokenizerFast:
...@@ -115,8 +115,8 @@ class DPRReaderTokenizerFast: ...@@ -115,8 +115,8 @@ class DPRReaderTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class ElectraTokenizerFast: class ElectraTokenizerFast:
...@@ -124,8 +124,8 @@ class ElectraTokenizerFast: ...@@ -124,8 +124,8 @@ class ElectraTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class FunnelTokenizerFast: class FunnelTokenizerFast:
...@@ -133,8 +133,8 @@ class FunnelTokenizerFast: ...@@ -133,8 +133,8 @@ class FunnelTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class GPT2TokenizerFast: class GPT2TokenizerFast:
...@@ -142,8 +142,8 @@ class GPT2TokenizerFast: ...@@ -142,8 +142,8 @@ class GPT2TokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class HerbertTokenizerFast: class HerbertTokenizerFast:
...@@ -151,8 +151,8 @@ class HerbertTokenizerFast: ...@@ -151,8 +151,8 @@ class HerbertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class LayoutLMTokenizerFast: class LayoutLMTokenizerFast:
...@@ -160,8 +160,8 @@ class LayoutLMTokenizerFast: ...@@ -160,8 +160,8 @@ class LayoutLMTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class LEDTokenizerFast: class LEDTokenizerFast:
...@@ -169,8 +169,8 @@ class LEDTokenizerFast: ...@@ -169,8 +169,8 @@ class LEDTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class LongformerTokenizerFast: class LongformerTokenizerFast:
...@@ -178,8 +178,8 @@ class LongformerTokenizerFast: ...@@ -178,8 +178,8 @@ class LongformerTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class LxmertTokenizerFast: class LxmertTokenizerFast:
...@@ -187,8 +187,8 @@ class LxmertTokenizerFast: ...@@ -187,8 +187,8 @@ class LxmertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class MBart50TokenizerFast: class MBart50TokenizerFast:
...@@ -196,8 +196,8 @@ class MBart50TokenizerFast: ...@@ -196,8 +196,8 @@ class MBart50TokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class MBartTokenizerFast: class MBartTokenizerFast:
...@@ -205,8 +205,8 @@ class MBartTokenizerFast: ...@@ -205,8 +205,8 @@ class MBartTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class MobileBertTokenizerFast: class MobileBertTokenizerFast:
...@@ -214,8 +214,8 @@ class MobileBertTokenizerFast: ...@@ -214,8 +214,8 @@ class MobileBertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class MPNetTokenizerFast: class MPNetTokenizerFast:
...@@ -223,8 +223,8 @@ class MPNetTokenizerFast: ...@@ -223,8 +223,8 @@ class MPNetTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class MT5TokenizerFast: class MT5TokenizerFast:
...@@ -232,8 +232,8 @@ class MT5TokenizerFast: ...@@ -232,8 +232,8 @@ class MT5TokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class OpenAIGPTTokenizerFast: class OpenAIGPTTokenizerFast:
...@@ -241,8 +241,8 @@ class OpenAIGPTTokenizerFast: ...@@ -241,8 +241,8 @@ class OpenAIGPTTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class PegasusTokenizerFast: class PegasusTokenizerFast:
...@@ -250,8 +250,8 @@ class PegasusTokenizerFast: ...@@ -250,8 +250,8 @@ class PegasusTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class ReformerTokenizerFast: class ReformerTokenizerFast:
...@@ -259,8 +259,8 @@ class ReformerTokenizerFast: ...@@ -259,8 +259,8 @@ class ReformerTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class RetriBertTokenizerFast: class RetriBertTokenizerFast:
...@@ -268,8 +268,8 @@ class RetriBertTokenizerFast: ...@@ -268,8 +268,8 @@ class RetriBertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class RobertaTokenizerFast: class RobertaTokenizerFast:
...@@ -277,8 +277,8 @@ class RobertaTokenizerFast: ...@@ -277,8 +277,8 @@ class RobertaTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class RoFormerTokenizerFast: class RoFormerTokenizerFast:
...@@ -286,8 +286,8 @@ class RoFormerTokenizerFast: ...@@ -286,8 +286,8 @@ class RoFormerTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class SqueezeBertTokenizerFast: class SqueezeBertTokenizerFast:
...@@ -295,8 +295,8 @@ class SqueezeBertTokenizerFast: ...@@ -295,8 +295,8 @@ class SqueezeBertTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class T5TokenizerFast: class T5TokenizerFast:
...@@ -304,8 +304,8 @@ class T5TokenizerFast: ...@@ -304,8 +304,8 @@ class T5TokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class XLMRobertaTokenizerFast: class XLMRobertaTokenizerFast:
...@@ -313,8 +313,8 @@ class XLMRobertaTokenizerFast: ...@@ -313,8 +313,8 @@ class XLMRobertaTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class XLNetTokenizerFast: class XLNetTokenizerFast:
...@@ -322,8 +322,8 @@ class XLNetTokenizerFast: ...@@ -322,8 +322,8 @@ class XLNetTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
class PreTrainedTokenizerFast: class PreTrainedTokenizerFast:
...@@ -331,5 +331,5 @@ class PreTrainedTokenizerFast: ...@@ -331,5 +331,5 @@ class PreTrainedTokenizerFast:
requires_backends(self, ["tokenizers"]) requires_backends(self, ["tokenizers"])
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, ["tokenizers"]) requires_backends(cls, ["tokenizers"])
utils/check_dummies.py
View file @ 3b1f5caf
...@@ -39,8 +39,8 @@ class {0}: ...@@ -39,8 +39,8 @@ class {0}:
requires_backends(self, {1}) requires_backends(self, {1})
@classmethod @classmethod
def from_pretrained(self, *args, **kwargs): def from_pretrained(cls, *args, **kwargs):
requires_backends(self, {1}) requires_backends(cls, {1})
""" """
DUMMY_CLASS = """ DUMMY_CLASS = """
...@@ -103,11 +103,14 @@ def read_init(): ...@@ -103,11 +103,14 @@ 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 = [ _pretrained = [
"Config" "ForCausalLM", "Config",
"ForCausalLM",
"ForConditionalGeneration", "ForConditionalGeneration",
"ForMaskedLM", "ForMaskedLM",
"ForMultipleChoice", "ForMultipleChoice",
"ForObjectDetection",
"ForQuestionAnswering", "ForQuestionAnswering",
"ForSegmentation",
"ForSequenceClassification", "ForSequenceClassification",
"ForTokenClassification", "ForTokenClassification",
"Model", "Model",
......
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