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

Add from_pretrained to dummy timm objects (#12097)

Browse Source 
* Add from_pretrained to dummy timm

* Fix at the source

* Update utils/check_dummies.py

Co-authored-by: Lysandre Debut <lysandre@huggingface.co>

* Missing pretrained dummies

* Style

Co-authored-by: Sylvain Gugger <sylvain.gugger@gmail.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
This commit is contained in:
Lysandre Debut
2021-06-11 18:27:10 +02:00
committed by GitHub
parent 15b498f3b8
commit 3b1f5caff2
8 changed files with 1158 additions and 1071 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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

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

File diff suppressed because it is too large Load Diff

68
src/transformers/utils/dummy_sentencepiece_objects.py
View File

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

680
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

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

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

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

148
src/transformers/utils/dummy_tokenizers_objects.py
View File

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

9
utils/check_dummies.py
View File

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