From 3b1f5caff26c08dfb74a76de1163f4becde9e828 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Fri, 11 Jun 2021 18:27:10 +0200 Subject: [PATCH] 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: Lysandre Debut * Missing pretrained dummies * Style Co-authored-by: Sylvain Gugger Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> --- src/transformers/utils/dummy_flax_objects.py | 152 +-- src/transformers/utils/dummy_pt_objects.py | 1148 +++++++++-------- .../utils/dummy_sentencepiece_objects.py | 68 +- src/transformers/utils/dummy_tf_objects.py | 680 +++++----- .../utils/dummy_timm_and_vision_objects.py | 12 +- src/transformers/utils/dummy_timm_objects.py | 12 +- .../utils/dummy_tokenizers_objects.py | 148 +-- utils/check_dummies.py | 9 +- 8 files changed, 1158 insertions(+), 1071 deletions(-) diff --git a/src/transformers/utils/dummy_flax_objects.py b/src/transformers/utils/dummy_flax_objects.py index 05c6b41f96..9907874abb 100644 --- a/src/transformers/utils/dummy_flax_objects.py +++ b/src/transformers/utils/dummy_flax_objects.py @@ -37,8 +37,8 @@ class FlaxPreTrainedModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) FLAX_MODEL_FOR_CAUSAL_LM_MAPPING = None @@ -73,8 +73,8 @@ class FlaxAutoModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxAutoModelForCausalLM: @@ -82,8 +82,8 @@ class FlaxAutoModelForCausalLM: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxAutoModelForMaskedLM: @@ -91,8 +91,8 @@ class FlaxAutoModelForMaskedLM: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxAutoModelForMultipleChoice: @@ -100,8 +100,8 @@ class FlaxAutoModelForMultipleChoice: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxAutoModelForNextSentencePrediction: @@ -109,8 +109,8 @@ class FlaxAutoModelForNextSentencePrediction: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxAutoModelForPreTraining: @@ -118,8 +118,8 @@ class FlaxAutoModelForPreTraining: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxAutoModelForQuestionAnswering: @@ -127,8 +127,8 @@ class FlaxAutoModelForQuestionAnswering: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxAutoModelForSequenceClassification: @@ -136,8 +136,8 @@ class FlaxAutoModelForSequenceClassification: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxAutoModelForTokenClassification: @@ -145,8 +145,8 @@ class FlaxAutoModelForTokenClassification: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxBertForMaskedLM: @@ -154,8 +154,8 @@ class FlaxBertForMaskedLM: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxBertForMultipleChoice: @@ -163,8 +163,8 @@ class FlaxBertForMultipleChoice: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxBertForNextSentencePrediction: @@ -182,8 +182,8 @@ class FlaxBertForQuestionAnswering: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxBertForSequenceClassification: @@ -191,8 +191,8 @@ class FlaxBertForSequenceClassification: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxBertForTokenClassification: @@ -200,8 +200,8 @@ class FlaxBertForTokenClassification: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxBertModel: @@ -209,8 +209,8 @@ class FlaxBertModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxBertPreTrainedModel: @@ -218,8 +218,8 @@ class FlaxBertPreTrainedModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxCLIPModel: @@ -227,8 +227,8 @@ class FlaxCLIPModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxCLIPPreTrainedModel: @@ -236,8 +236,8 @@ class FlaxCLIPPreTrainedModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxCLIPTextModel: @@ -245,8 +245,8 @@ class FlaxCLIPTextModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxCLIPVisionModel: @@ -254,8 +254,8 @@ class FlaxCLIPVisionModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxElectraForMaskedLM: @@ -263,8 +263,8 @@ class FlaxElectraForMaskedLM: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxElectraForMultipleChoice: @@ -272,8 +272,8 @@ class FlaxElectraForMultipleChoice: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxElectraForPreTraining: @@ -286,8 +286,8 @@ class FlaxElectraForQuestionAnswering: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxElectraForSequenceClassification: @@ -295,8 +295,8 @@ class FlaxElectraForSequenceClassification: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxElectraForTokenClassification: @@ -304,8 +304,8 @@ class FlaxElectraForTokenClassification: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxElectraModel: @@ -313,8 +313,8 @@ class FlaxElectraModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxElectraPreTrainedModel: @@ -322,8 +322,8 @@ class FlaxElectraPreTrainedModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxGPT2LMHeadModel: @@ -331,8 +331,8 @@ class FlaxGPT2LMHeadModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxGPT2Model: @@ -340,8 +340,8 @@ class FlaxGPT2Model: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxRobertaForMaskedLM: @@ -349,8 +349,8 @@ class FlaxRobertaForMaskedLM: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxRobertaForMultipleChoice: @@ -358,8 +358,8 @@ class FlaxRobertaForMultipleChoice: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxRobertaForQuestionAnswering: @@ -367,8 +367,8 @@ class FlaxRobertaForQuestionAnswering: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxRobertaForSequenceClassification: @@ -376,8 +376,8 @@ class FlaxRobertaForSequenceClassification: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxRobertaForTokenClassification: @@ -385,8 +385,8 @@ class FlaxRobertaForTokenClassification: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxRobertaModel: @@ -394,8 +394,8 @@ class FlaxRobertaModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxRobertaPreTrainedModel: @@ -403,8 +403,8 @@ class FlaxRobertaPreTrainedModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) class FlaxViTForImageClassification: @@ -417,5 +417,5 @@ class FlaxViTModel: requires_backends(self, ["flax"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["flax"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["flax"]) diff --git a/src/transformers/utils/dummy_pt_objects.py b/src/transformers/utils/dummy_pt_objects.py index f3b8e81348..1fa3f30cf5 100644 --- a/src/transformers/utils/dummy_pt_objects.py +++ b/src/transformers/utils/dummy_pt_objects.py @@ -22,8 +22,8 @@ class DataCollatorForLanguageModeling: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DataCollatorForPermutationLanguageModeling: @@ -31,8 +31,8 @@ class DataCollatorForPermutationLanguageModeling: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DataCollatorForSeq2Seq: @@ -50,8 +50,8 @@ class DataCollatorForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DataCollatorForWholeWordMask: @@ -232,8 +232,8 @@ class PreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def apply_chunking_to_forward(*args, **kwargs): @@ -252,8 +252,8 @@ class AlbertForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AlbertForMultipleChoice: @@ -261,8 +261,8 @@ class AlbertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AlbertForPreTraining: @@ -275,8 +275,8 @@ class AlbertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AlbertForSequenceClassification: @@ -284,8 +284,8 @@ class AlbertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AlbertForTokenClassification: @@ -293,8 +293,8 @@ class AlbertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AlbertModel: @@ -302,8 +302,8 @@ class AlbertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AlbertPreTrainedModel: @@ -311,8 +311,8 @@ class AlbertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_albert(*args, **kwargs): @@ -366,8 +366,8 @@ class AutoModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForCausalLM: @@ -375,8 +375,8 @@ class AutoModelForCausalLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForImageClassification: @@ -384,8 +384,8 @@ class AutoModelForImageClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForMaskedLM: @@ -393,8 +393,8 @@ class AutoModelForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForMultipleChoice: @@ -402,8 +402,8 @@ class AutoModelForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForNextSentencePrediction: @@ -411,8 +411,8 @@ class AutoModelForNextSentencePrediction: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForPreTraining: @@ -420,8 +420,8 @@ class AutoModelForPreTraining: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForQuestionAnswering: @@ -429,8 +429,8 @@ class AutoModelForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForSeq2SeqLM: @@ -438,8 +438,8 @@ class AutoModelForSeq2SeqLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForSequenceClassification: @@ -447,8 +447,8 @@ class AutoModelForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForTableQuestionAnswering: @@ -456,8 +456,8 @@ class AutoModelForTableQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelForTokenClassification: @@ -465,8 +465,8 @@ class AutoModelForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class AutoModelWithLMHead: @@ -474,8 +474,8 @@ class AutoModelWithLMHead: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) BART_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -485,14 +485,18 @@ class BartForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class BartForConditionalGeneration: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BartForQuestionAnswering: @@ -500,8 +504,8 @@ class BartForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BartForSequenceClassification: @@ -509,8 +513,8 @@ class BartForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BartModel: @@ -518,8 +522,8 @@ class BartModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BartPretrainedModel: @@ -527,8 +531,8 @@ class BartPretrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class PretrainedBartModel: @@ -536,8 +540,8 @@ class PretrainedBartModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -548,8 +552,8 @@ class BertForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BertForMultipleChoice: @@ -557,8 +561,8 @@ class BertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BertForNextSentencePrediction: @@ -576,8 +580,8 @@ class BertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BertForSequenceClassification: @@ -585,8 +589,8 @@ class BertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BertForTokenClassification: @@ -594,8 +598,8 @@ class BertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BertLayer: @@ -608,8 +612,8 @@ class BertLMHeadModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BertModel: @@ -617,8 +621,8 @@ class BertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BertPreTrainedModel: @@ -626,8 +630,8 @@ class BertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_bert(*args, **kwargs): @@ -655,14 +659,18 @@ class BigBirdForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class BigBirdForMaskedLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BigBirdForMultipleChoice: @@ -670,8 +678,8 @@ class BigBirdForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BigBirdForPreTraining: @@ -684,8 +692,8 @@ class BigBirdForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BigBirdForSequenceClassification: @@ -693,8 +701,8 @@ class BigBirdForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BigBirdForTokenClassification: @@ -702,8 +710,8 @@ class BigBirdForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BigBirdLayer: @@ -716,8 +724,8 @@ class BigBirdModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BigBirdPreTrainedModel: @@ -725,8 +733,8 @@ class BigBirdPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_big_bird(*args, **kwargs): @@ -740,14 +748,18 @@ class BigBirdPegasusForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class BigBirdPegasusForConditionalGeneration: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BigBirdPegasusForQuestionAnswering: @@ -755,8 +767,8 @@ class BigBirdPegasusForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BigBirdPegasusForSequenceClassification: @@ -764,8 +776,8 @@ class BigBirdPegasusForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BigBirdPegasusModel: @@ -773,8 +785,8 @@ class BigBirdPegasusModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) BLENDERBOT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -784,14 +796,18 @@ class BlenderbotForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class BlenderbotForConditionalGeneration: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BlenderbotModel: @@ -799,8 +815,8 @@ class BlenderbotModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) BLENDERBOT_SMALL_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -810,14 +826,18 @@ class BlenderbotSmallForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class BlenderbotSmallForConditionalGeneration: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class BlenderbotSmallModel: @@ -825,8 +845,8 @@ class BlenderbotSmallModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -836,14 +856,18 @@ class CamembertForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class CamembertForMaskedLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CamembertForMultipleChoice: @@ -851,8 +875,8 @@ class CamembertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CamembertForQuestionAnswering: @@ -860,8 +884,8 @@ class CamembertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CamembertForSequenceClassification: @@ -869,8 +893,8 @@ class CamembertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CamembertForTokenClassification: @@ -878,8 +902,8 @@ class CamembertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CamembertModel: @@ -887,8 +911,8 @@ class CamembertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) CLIP_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -899,8 +923,8 @@ class CLIPModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CLIPPreTrainedModel: @@ -908,8 +932,8 @@ class CLIPPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CLIPTextModel: @@ -917,8 +941,8 @@ class CLIPTextModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CLIPVisionModel: @@ -926,8 +950,8 @@ class CLIPVisionModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) CONVBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -938,8 +962,8 @@ class ConvBertForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ConvBertForMultipleChoice: @@ -947,8 +971,8 @@ class ConvBertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ConvBertForQuestionAnswering: @@ -956,8 +980,8 @@ class ConvBertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ConvBertForSequenceClassification: @@ -965,8 +989,8 @@ class ConvBertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ConvBertForTokenClassification: @@ -974,8 +998,8 @@ class ConvBertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ConvBertLayer: @@ -988,8 +1012,8 @@ class ConvBertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ConvBertPreTrainedModel: @@ -997,8 +1021,8 @@ class ConvBertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_convbert(*args, **kwargs): @@ -1013,8 +1037,8 @@ class CTRLForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CTRLLMHeadModel: @@ -1022,8 +1046,8 @@ class CTRLLMHeadModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CTRLModel: @@ -1031,8 +1055,8 @@ class CTRLModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class CTRLPreTrainedModel: @@ -1040,8 +1064,8 @@ class CTRLPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) DEBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1052,8 +1076,8 @@ class DebertaForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaForQuestionAnswering: @@ -1061,8 +1085,8 @@ class DebertaForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaForSequenceClassification: @@ -1070,8 +1094,8 @@ class DebertaForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaForTokenClassification: @@ -1079,8 +1103,8 @@ class DebertaForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaModel: @@ -1088,8 +1112,8 @@ class DebertaModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaPreTrainedModel: @@ -1097,8 +1121,8 @@ class DebertaPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) DEBERTA_V2_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1109,8 +1133,8 @@ class DebertaV2ForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaV2ForQuestionAnswering: @@ -1118,8 +1142,8 @@ class DebertaV2ForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaV2ForSequenceClassification: @@ -1127,8 +1151,8 @@ class DebertaV2ForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaV2ForTokenClassification: @@ -1136,8 +1160,8 @@ class DebertaV2ForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaV2Model: @@ -1145,8 +1169,8 @@ class DebertaV2Model: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DebertaV2PreTrainedModel: @@ -1154,8 +1178,8 @@ class DebertaV2PreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) DEIT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1176,8 +1200,8 @@ class DeiTModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DeiTPreTrainedModel: @@ -1185,8 +1209,8 @@ class DeiTPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1197,8 +1221,8 @@ class DistilBertForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DistilBertForMultipleChoice: @@ -1206,8 +1230,8 @@ class DistilBertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DistilBertForQuestionAnswering: @@ -1215,8 +1239,8 @@ class DistilBertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DistilBertForSequenceClassification: @@ -1224,8 +1248,8 @@ class DistilBertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DistilBertForTokenClassification: @@ -1233,8 +1257,8 @@ class DistilBertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DistilBertModel: @@ -1242,8 +1266,8 @@ class DistilBertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class DistilBertPreTrainedModel: @@ -1251,8 +1275,8 @@ class DistilBertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1302,8 +1326,8 @@ class ElectraForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ElectraForMultipleChoice: @@ -1311,8 +1335,8 @@ class ElectraForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ElectraForPreTraining: @@ -1325,8 +1349,8 @@ class ElectraForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ElectraForSequenceClassification: @@ -1334,8 +1358,8 @@ class ElectraForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ElectraForTokenClassification: @@ -1343,8 +1367,8 @@ class ElectraForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ElectraModel: @@ -1352,8 +1376,8 @@ class ElectraModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ElectraPreTrainedModel: @@ -1361,8 +1385,8 @@ class ElectraPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_electra(*args, **kwargs): @@ -1374,8 +1398,8 @@ class EncoderDecoderModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1386,8 +1410,8 @@ class FlaubertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FlaubertForQuestionAnswering: @@ -1395,8 +1419,8 @@ class FlaubertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FlaubertForQuestionAnsweringSimple: @@ -1404,8 +1428,8 @@ class FlaubertForQuestionAnsweringSimple: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FlaubertForSequenceClassification: @@ -1413,8 +1437,8 @@ class FlaubertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FlaubertForTokenClassification: @@ -1422,8 +1446,8 @@ class FlaubertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FlaubertModel: @@ -1431,8 +1455,8 @@ class FlaubertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FlaubertWithLMHeadModel: @@ -1440,8 +1464,8 @@ class FlaubertWithLMHeadModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FSMTForConditionalGeneration: @@ -1449,8 +1473,8 @@ class FSMTForConditionalGeneration: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FSMTModel: @@ -1458,8 +1482,8 @@ class FSMTModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class PretrainedFSMTModel: @@ -1467,8 +1491,8 @@ class PretrainedFSMTModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) FUNNEL_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1479,8 +1503,8 @@ class FunnelBaseModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FunnelForMaskedLM: @@ -1488,8 +1512,8 @@ class FunnelForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FunnelForMultipleChoice: @@ -1497,8 +1521,8 @@ class FunnelForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FunnelForPreTraining: @@ -1511,8 +1535,8 @@ class FunnelForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FunnelForSequenceClassification: @@ -1520,8 +1544,8 @@ class FunnelForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FunnelForTokenClassification: @@ -1529,8 +1553,8 @@ class FunnelForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class FunnelModel: @@ -1538,8 +1562,8 @@ class FunnelModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_funnel(*args, **kwargs): @@ -1554,8 +1578,8 @@ class GPT2DoubleHeadsModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class GPT2ForSequenceClassification: @@ -1563,8 +1587,8 @@ class GPT2ForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class GPT2LMHeadModel: @@ -1572,8 +1596,8 @@ class GPT2LMHeadModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class GPT2Model: @@ -1581,8 +1605,8 @@ class GPT2Model: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class GPT2PreTrainedModel: @@ -1590,8 +1614,8 @@ class GPT2PreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_gpt2(*args, **kwargs): @@ -1605,14 +1629,18 @@ class GPTNeoForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class GPTNeoForSequenceClassification: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class GPTNeoModel: @@ -1620,8 +1648,8 @@ class GPTNeoModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class GPTNeoPreTrainedModel: @@ -1629,8 +1657,8 @@ class GPTNeoPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_gpt_neo(*args, **kwargs): @@ -1645,8 +1673,8 @@ class IBertForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class IBertForMultipleChoice: @@ -1654,8 +1682,8 @@ class IBertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class IBertForQuestionAnswering: @@ -1663,8 +1691,8 @@ class IBertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class IBertForSequenceClassification: @@ -1672,8 +1700,8 @@ class IBertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class IBertForTokenClassification: @@ -1681,8 +1709,8 @@ class IBertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class IBertModel: @@ -1690,8 +1718,8 @@ class IBertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class IBertPreTrainedModel: @@ -1699,8 +1727,8 @@ class IBertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1711,8 +1739,8 @@ class LayoutLMForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LayoutLMForSequenceClassification: @@ -1720,8 +1748,8 @@ class LayoutLMForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LayoutLMForTokenClassification: @@ -1729,8 +1757,8 @@ class LayoutLMForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LayoutLMModel: @@ -1738,8 +1766,8 @@ class LayoutLMModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) LED_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1750,8 +1778,8 @@ class LEDForConditionalGeneration: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LEDForQuestionAnswering: @@ -1759,8 +1787,8 @@ class LEDForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LEDForSequenceClassification: @@ -1768,8 +1796,8 @@ class LEDForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LEDModel: @@ -1777,8 +1805,8 @@ class LEDModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1789,8 +1817,8 @@ class LongformerForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LongformerForMultipleChoice: @@ -1798,8 +1826,8 @@ class LongformerForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LongformerForQuestionAnswering: @@ -1807,8 +1835,8 @@ class LongformerForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LongformerForSequenceClassification: @@ -1816,8 +1844,8 @@ class LongformerForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LongformerForTokenClassification: @@ -1825,8 +1853,8 @@ class LongformerForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LongformerModel: @@ -1834,8 +1862,8 @@ class LongformerModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LongformerSelfAttention: @@ -1866,8 +1894,8 @@ class LukeModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LukePreTrainedModel: @@ -1875,8 +1903,8 @@ class LukePreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LxmertEncoder: @@ -1894,8 +1922,8 @@ class LxmertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LxmertModel: @@ -1903,8 +1931,8 @@ class LxmertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LxmertPreTrainedModel: @@ -1912,8 +1940,8 @@ class LxmertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class LxmertVisualFeatureEncoder: @@ -1934,8 +1962,8 @@ class M2M100ForConditionalGeneration: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class M2M100Model: @@ -1943,22 +1971,26 @@ class M2M100Model: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MarianForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class MarianModel: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MarianMTModel: @@ -1966,22 +1998,26 @@ class MarianMTModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MBartForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class MBartForConditionalGeneration: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MBartForQuestionAnswering: @@ -1989,8 +2025,8 @@ class MBartForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MBartForSequenceClassification: @@ -1998,8 +2034,8 @@ class MBartForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MBartModel: @@ -2007,8 +2043,8 @@ class MBartModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2018,14 +2054,18 @@ class MegatronBertForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class MegatronBertForMaskedLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MegatronBertForMultipleChoice: @@ -2033,8 +2073,8 @@ class MegatronBertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MegatronBertForNextSentencePrediction: @@ -2052,8 +2092,8 @@ class MegatronBertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MegatronBertForSequenceClassification: @@ -2061,8 +2101,8 @@ class MegatronBertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MegatronBertForTokenClassification: @@ -2070,8 +2110,8 @@ class MegatronBertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MegatronBertModel: @@ -2079,8 +2119,8 @@ class MegatronBertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MMBTForClassification: @@ -2093,8 +2133,8 @@ class MMBTModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ModalEmbeddings: @@ -2110,8 +2150,8 @@ class MobileBertForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MobileBertForMultipleChoice: @@ -2119,8 +2159,8 @@ class MobileBertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MobileBertForNextSentencePrediction: @@ -2138,8 +2178,8 @@ class MobileBertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MobileBertForSequenceClassification: @@ -2147,8 +2187,8 @@ class MobileBertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MobileBertForTokenClassification: @@ -2156,8 +2196,8 @@ class MobileBertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MobileBertLayer: @@ -2170,8 +2210,8 @@ class MobileBertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MobileBertPreTrainedModel: @@ -2179,8 +2219,8 @@ class MobileBertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_mobilebert(*args, **kwargs): @@ -2195,8 +2235,8 @@ class MPNetForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MPNetForMultipleChoice: @@ -2204,8 +2244,8 @@ class MPNetForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MPNetForQuestionAnswering: @@ -2213,8 +2253,8 @@ class MPNetForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MPNetForSequenceClassification: @@ -2222,8 +2262,8 @@ class MPNetForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MPNetForTokenClassification: @@ -2231,8 +2271,8 @@ class MPNetForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MPNetLayer: @@ -2245,8 +2285,8 @@ class MPNetModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MPNetPreTrainedModel: @@ -2254,8 +2294,8 @@ class MPNetPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MT5EncoderModel: @@ -2263,8 +2303,8 @@ class MT5EncoderModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MT5ForConditionalGeneration: @@ -2272,8 +2312,8 @@ class MT5ForConditionalGeneration: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class MT5Model: @@ -2281,8 +2321,8 @@ class MT5Model: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2293,8 +2333,8 @@ class OpenAIGPTDoubleHeadsModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class OpenAIGPTForSequenceClassification: @@ -2302,8 +2342,8 @@ class OpenAIGPTForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class OpenAIGPTLMHeadModel: @@ -2311,8 +2351,8 @@ class OpenAIGPTLMHeadModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class OpenAIGPTModel: @@ -2320,8 +2360,8 @@ class OpenAIGPTModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class OpenAIGPTPreTrainedModel: @@ -2329,8 +2369,8 @@ class OpenAIGPTPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_openai_gpt(*args, **kwargs): @@ -2341,14 +2381,18 @@ class PegasusForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class PegasusForConditionalGeneration: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class PegasusModel: @@ -2356,8 +2400,8 @@ class PegasusModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2377,14 +2421,18 @@ class ProphetNetForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class ProphetNetForConditionalGeneration: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ProphetNetModel: @@ -2392,8 +2440,8 @@ class ProphetNetModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ProphetNetPreTrainedModel: @@ -2401,8 +2449,8 @@ class ProphetNetPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RagModel: @@ -2410,8 +2458,8 @@ class RagModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RagSequenceForGeneration: @@ -2437,8 +2485,8 @@ class ReformerForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ReformerForQuestionAnswering: @@ -2446,8 +2494,8 @@ class ReformerForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ReformerForSequenceClassification: @@ -2455,8 +2503,8 @@ class ReformerForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ReformerLayer: @@ -2469,8 +2517,8 @@ class ReformerModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ReformerModelWithLMHead: @@ -2478,8 +2526,8 @@ class ReformerModelWithLMHead: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) RETRIBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2490,8 +2538,8 @@ class RetriBertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RetriBertPreTrainedModel: @@ -2499,8 +2547,8 @@ class RetriBertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2510,14 +2558,18 @@ class RobertaForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class RobertaForMaskedLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RobertaForMultipleChoice: @@ -2525,8 +2577,8 @@ class RobertaForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RobertaForQuestionAnswering: @@ -2534,8 +2586,8 @@ class RobertaForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RobertaForSequenceClassification: @@ -2543,8 +2595,8 @@ class RobertaForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RobertaForTokenClassification: @@ -2552,8 +2604,8 @@ class RobertaForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RobertaModel: @@ -2561,8 +2613,8 @@ class RobertaModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) ROFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2572,14 +2624,18 @@ class RoFormerForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class RoFormerForMaskedLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RoFormerForMultipleChoice: @@ -2587,8 +2643,8 @@ class RoFormerForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RoFormerForQuestionAnswering: @@ -2596,8 +2652,8 @@ class RoFormerForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RoFormerForSequenceClassification: @@ -2605,8 +2661,8 @@ class RoFormerForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RoFormerForTokenClassification: @@ -2614,8 +2670,8 @@ class RoFormerForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RoFormerLayer: @@ -2628,8 +2684,8 @@ class RoFormerModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class RoFormerPreTrainedModel: @@ -2637,8 +2693,8 @@ class RoFormerPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_roformer(*args, **kwargs): @@ -2653,8 +2709,8 @@ class Speech2TextForConditionalGeneration: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class Speech2TextModel: @@ -2662,8 +2718,8 @@ class Speech2TextModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) SQUEEZEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2674,8 +2730,8 @@ class SqueezeBertForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class SqueezeBertForMultipleChoice: @@ -2683,8 +2739,8 @@ class SqueezeBertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class SqueezeBertForQuestionAnswering: @@ -2692,8 +2748,8 @@ class SqueezeBertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class SqueezeBertForSequenceClassification: @@ -2701,8 +2757,8 @@ class SqueezeBertForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class SqueezeBertForTokenClassification: @@ -2710,8 +2766,8 @@ class SqueezeBertForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class SqueezeBertModel: @@ -2719,8 +2775,8 @@ class SqueezeBertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class SqueezeBertModule: @@ -2733,8 +2789,8 @@ class SqueezeBertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) T5_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2745,8 +2801,8 @@ class T5EncoderModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class T5ForConditionalGeneration: @@ -2754,8 +2810,8 @@ class T5ForConditionalGeneration: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class T5Model: @@ -2763,8 +2819,8 @@ class T5Model: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class T5PreTrainedModel: @@ -2772,8 +2828,8 @@ class T5PreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_t5(*args, **kwargs): @@ -2788,8 +2844,8 @@ class TapasForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class TapasForQuestionAnswering: @@ -2797,8 +2853,8 @@ class TapasForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class TapasForSequenceClassification: @@ -2806,8 +2862,8 @@ class TapasForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class TapasModel: @@ -2815,8 +2871,8 @@ class TapasModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2832,8 +2888,8 @@ class TransfoXLForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class TransfoXLLMHeadModel: @@ -2841,8 +2897,8 @@ class TransfoXLLMHeadModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class TransfoXLModel: @@ -2850,8 +2906,8 @@ class TransfoXLModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class TransfoXLPreTrainedModel: @@ -2859,8 +2915,8 @@ class TransfoXLPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_transfo_xl(*args, **kwargs): @@ -2875,8 +2931,8 @@ class VisualBertForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class VisualBertForPreTraining: @@ -2889,8 +2945,8 @@ class VisualBertForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class VisualBertForRegionToPhraseAlignment: @@ -2913,8 +2969,8 @@ class VisualBertModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class VisualBertPreTrainedModel: @@ -2922,8 +2978,8 @@ class VisualBertPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) VIT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2939,8 +2995,8 @@ class ViTModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class ViTPreTrainedModel: @@ -2948,8 +3004,8 @@ class ViTPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -2965,8 +3021,8 @@ class Wav2Vec2ForMaskedLM: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class Wav2Vec2ForPreTraining: @@ -2979,8 +3035,8 @@ class Wav2Vec2Model: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class Wav2Vec2PreTrainedModel: @@ -2988,8 +3044,8 @@ class Wav2Vec2PreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -3000,8 +3056,8 @@ class XLMForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMForQuestionAnswering: @@ -3009,8 +3065,8 @@ class XLMForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMForQuestionAnsweringSimple: @@ -3018,8 +3074,8 @@ class XLMForQuestionAnsweringSimple: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMForSequenceClassification: @@ -3027,8 +3083,8 @@ class XLMForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMForTokenClassification: @@ -3036,8 +3092,8 @@ class XLMForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMModel: @@ -3045,8 +3101,8 @@ class XLMModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMPreTrainedModel: @@ -3054,8 +3110,8 @@ class XLMPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMWithLMHeadModel: @@ -3063,8 +3119,8 @@ class XLMWithLMHeadModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) XLM_PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -3084,14 +3140,18 @@ class XLMProphetNetForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class XLMProphetNetForConditionalGeneration: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMProphetNetModel: @@ -3099,8 +3159,8 @@ class XLMProphetNetModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -3110,14 +3170,18 @@ class XLMRobertaForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) + class XLMRobertaForMaskedLM: def __init__(self, *args, **kwargs): requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMRobertaForMultipleChoice: @@ -3125,8 +3189,8 @@ class XLMRobertaForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMRobertaForQuestionAnswering: @@ -3134,8 +3198,8 @@ class XLMRobertaForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMRobertaForSequenceClassification: @@ -3143,8 +3207,8 @@ class XLMRobertaForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMRobertaForTokenClassification: @@ -3152,8 +3216,8 @@ class XLMRobertaForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLMRobertaModel: @@ -3161,8 +3225,8 @@ class XLMRobertaModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) XLNET_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -3173,8 +3237,8 @@ class XLNetForMultipleChoice: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLNetForQuestionAnswering: @@ -3182,8 +3246,8 @@ class XLNetForQuestionAnswering: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLNetForQuestionAnsweringSimple: @@ -3191,8 +3255,8 @@ class XLNetForQuestionAnsweringSimple: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLNetForSequenceClassification: @@ -3200,8 +3264,8 @@ class XLNetForSequenceClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLNetForTokenClassification: @@ -3209,8 +3273,8 @@ class XLNetForTokenClassification: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLNetLMHeadModel: @@ -3218,8 +3282,8 @@ class XLNetLMHeadModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLNetModel: @@ -3227,8 +3291,8 @@ class XLNetModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) class XLNetPreTrainedModel: @@ -3236,8 +3300,8 @@ class XLNetPreTrainedModel: requires_backends(self, ["torch"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["torch"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch"]) def load_tf_weights_in_xlnet(*args, **kwargs): diff --git a/src/transformers/utils/dummy_sentencepiece_objects.py b/src/transformers/utils/dummy_sentencepiece_objects.py index d87263c8c7..ef2b167dff 100644 --- a/src/transformers/utils/dummy_sentencepiece_objects.py +++ b/src/transformers/utils/dummy_sentencepiece_objects.py @@ -7,8 +7,8 @@ class AlbertTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class BarthezTokenizer: @@ -16,8 +16,8 @@ class BarthezTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class BertGenerationTokenizer: @@ -25,8 +25,8 @@ class BertGenerationTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class CamembertTokenizer: @@ -34,8 +34,8 @@ class CamembertTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class DebertaV2Tokenizer: @@ -43,8 +43,8 @@ class DebertaV2Tokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class M2M100Tokenizer: @@ -52,8 +52,8 @@ class M2M100Tokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class MarianTokenizer: @@ -61,8 +61,8 @@ class MarianTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class MBart50Tokenizer: @@ -70,8 +70,8 @@ class MBart50Tokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class MBartTokenizer: @@ -79,8 +79,8 @@ class MBartTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class MT5Tokenizer: @@ -88,8 +88,8 @@ class MT5Tokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class PegasusTokenizer: @@ -97,8 +97,8 @@ class PegasusTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class ReformerTokenizer: @@ -106,8 +106,8 @@ class ReformerTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class Speech2TextTokenizer: @@ -115,8 +115,8 @@ class Speech2TextTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class T5Tokenizer: @@ -124,8 +124,8 @@ class T5Tokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class XLMProphetNetTokenizer: @@ -133,8 +133,8 @@ class XLMProphetNetTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class XLMRobertaTokenizer: @@ -142,8 +142,8 @@ class XLMRobertaTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) class XLNetTokenizer: @@ -151,5 +151,5 @@ class XLNetTokenizer: requires_backends(self, ["sentencepiece"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["sentencepiece"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["sentencepiece"]) diff --git a/src/transformers/utils/dummy_tf_objects.py b/src/transformers/utils/dummy_tf_objects.py index 380f297d78..33ad41f70a 100644 --- a/src/transformers/utils/dummy_tf_objects.py +++ b/src/transformers/utils/dummy_tf_objects.py @@ -24,8 +24,8 @@ class TFLayoutLMForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLayoutLMForSequenceClassification: @@ -33,8 +33,8 @@ class TFLayoutLMForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLayoutLMForTokenClassification: @@ -42,8 +42,8 @@ class TFLayoutLMForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLayoutLMMainLayer: @@ -56,8 +56,8 @@ class TFLayoutLMModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLayoutLMPreTrainedModel: @@ -65,8 +65,8 @@ class TFLayoutLMPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFPreTrainedModel: @@ -74,8 +74,8 @@ class TFPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFSequenceSummary: @@ -100,8 +100,8 @@ class TFAlbertForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAlbertForMultipleChoice: @@ -109,8 +109,8 @@ class TFAlbertForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAlbertForPreTraining: @@ -123,8 +123,8 @@ class TFAlbertForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAlbertForSequenceClassification: @@ -132,8 +132,8 @@ class TFAlbertForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAlbertForTokenClassification: @@ -141,8 +141,8 @@ class TFAlbertForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAlbertMainLayer: @@ -155,8 +155,8 @@ class TFAlbertModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAlbertPreTrainedModel: @@ -164,8 +164,8 @@ class TFAlbertPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_MODEL_FOR_CAUSAL_LM_MAPPING = None @@ -206,8 +206,8 @@ class TFAutoModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAutoModelForCausalLM: @@ -215,8 +215,8 @@ class TFAutoModelForCausalLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAutoModelForMaskedLM: @@ -224,8 +224,8 @@ class TFAutoModelForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAutoModelForMultipleChoice: @@ -233,8 +233,8 @@ class TFAutoModelForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAutoModelForPreTraining: @@ -242,8 +242,8 @@ class TFAutoModelForPreTraining: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAutoModelForQuestionAnswering: @@ -251,8 +251,8 @@ class TFAutoModelForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAutoModelForSeq2SeqLM: @@ -260,8 +260,8 @@ class TFAutoModelForSeq2SeqLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAutoModelForSequenceClassification: @@ -269,8 +269,8 @@ class TFAutoModelForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAutoModelForTokenClassification: @@ -278,8 +278,8 @@ class TFAutoModelForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFAutoModelWithLMHead: @@ -287,8 +287,8 @@ class TFAutoModelWithLMHead: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBartForConditionalGeneration: @@ -296,8 +296,8 @@ class TFBartForConditionalGeneration: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBartModel: @@ -305,8 +305,8 @@ class TFBartModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBartPretrainedModel: @@ -314,8 +314,8 @@ class TFBartPretrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -331,8 +331,8 @@ class TFBertForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBertForMultipleChoice: @@ -340,8 +340,8 @@ class TFBertForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBertForNextSentencePrediction: @@ -359,8 +359,8 @@ class TFBertForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBertForSequenceClassification: @@ -368,8 +368,8 @@ class TFBertForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBertForTokenClassification: @@ -377,8 +377,8 @@ class TFBertForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBertLMHeadModel: @@ -386,8 +386,8 @@ class TFBertLMHeadModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBertMainLayer: @@ -400,8 +400,8 @@ class TFBertModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBertPreTrainedModel: @@ -409,8 +409,8 @@ class TFBertPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBlenderbotForConditionalGeneration: @@ -418,8 +418,8 @@ class TFBlenderbotForConditionalGeneration: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBlenderbotModel: @@ -427,8 +427,8 @@ class TFBlenderbotModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBlenderbotSmallForConditionalGeneration: @@ -436,8 +436,8 @@ class TFBlenderbotSmallForConditionalGeneration: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFBlenderbotSmallModel: @@ -445,8 +445,8 @@ class TFBlenderbotSmallModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -457,8 +457,8 @@ class TFCamembertForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFCamembertForMultipleChoice: @@ -466,8 +466,8 @@ class TFCamembertForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFCamembertForQuestionAnswering: @@ -475,8 +475,8 @@ class TFCamembertForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFCamembertForSequenceClassification: @@ -484,8 +484,8 @@ class TFCamembertForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFCamembertForTokenClassification: @@ -493,8 +493,8 @@ class TFCamembertForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFCamembertModel: @@ -502,8 +502,8 @@ class TFCamembertModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_CONVBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -514,8 +514,8 @@ class TFConvBertForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFConvBertForMultipleChoice: @@ -523,8 +523,8 @@ class TFConvBertForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFConvBertForQuestionAnswering: @@ -532,8 +532,8 @@ class TFConvBertForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFConvBertForSequenceClassification: @@ -541,8 +541,8 @@ class TFConvBertForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFConvBertForTokenClassification: @@ -550,8 +550,8 @@ class TFConvBertForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFConvBertLayer: @@ -564,8 +564,8 @@ class TFConvBertModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFConvBertPreTrainedModel: @@ -573,8 +573,8 @@ class TFConvBertPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_CTRL_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -585,8 +585,8 @@ class TFCTRLForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFCTRLLMHeadModel: @@ -594,8 +594,8 @@ class TFCTRLLMHeadModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFCTRLModel: @@ -603,8 +603,8 @@ class TFCTRLModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFCTRLPreTrainedModel: @@ -612,8 +612,8 @@ class TFCTRLPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -624,8 +624,8 @@ class TFDistilBertForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFDistilBertForMultipleChoice: @@ -633,8 +633,8 @@ class TFDistilBertForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFDistilBertForQuestionAnswering: @@ -642,8 +642,8 @@ class TFDistilBertForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFDistilBertForSequenceClassification: @@ -651,8 +651,8 @@ class TFDistilBertForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFDistilBertForTokenClassification: @@ -660,8 +660,8 @@ class TFDistilBertForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFDistilBertMainLayer: @@ -674,8 +674,8 @@ class TFDistilBertModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFDistilBertPreTrainedModel: @@ -683,8 +683,8 @@ class TFDistilBertPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -734,8 +734,8 @@ class TFElectraForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFElectraForMultipleChoice: @@ -743,8 +743,8 @@ class TFElectraForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFElectraForPreTraining: @@ -757,8 +757,8 @@ class TFElectraForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFElectraForSequenceClassification: @@ -766,8 +766,8 @@ class TFElectraForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFElectraForTokenClassification: @@ -775,8 +775,8 @@ class TFElectraForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFElectraModel: @@ -784,8 +784,8 @@ class TFElectraModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFElectraPreTrainedModel: @@ -793,8 +793,8 @@ class TFElectraPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -805,8 +805,8 @@ class TFFlaubertForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFlaubertForQuestionAnsweringSimple: @@ -814,8 +814,8 @@ class TFFlaubertForQuestionAnsweringSimple: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFlaubertForSequenceClassification: @@ -823,8 +823,8 @@ class TFFlaubertForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFlaubertForTokenClassification: @@ -832,8 +832,8 @@ class TFFlaubertForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFlaubertModel: @@ -841,8 +841,8 @@ class TFFlaubertModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFlaubertWithLMHeadModel: @@ -850,8 +850,8 @@ class TFFlaubertWithLMHeadModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_FUNNEL_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -862,8 +862,8 @@ class TFFunnelBaseModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFunnelForMaskedLM: @@ -871,8 +871,8 @@ class TFFunnelForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFunnelForMultipleChoice: @@ -880,8 +880,8 @@ class TFFunnelForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFunnelForPreTraining: @@ -894,8 +894,8 @@ class TFFunnelForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFunnelForSequenceClassification: @@ -903,8 +903,8 @@ class TFFunnelForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFunnelForTokenClassification: @@ -912,8 +912,8 @@ class TFFunnelForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFFunnelModel: @@ -921,8 +921,8 @@ class TFFunnelModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -933,8 +933,8 @@ class TFGPT2DoubleHeadsModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFGPT2ForSequenceClassification: @@ -942,8 +942,8 @@ class TFGPT2ForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFGPT2LMHeadModel: @@ -951,8 +951,8 @@ class TFGPT2LMHeadModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFGPT2MainLayer: @@ -965,8 +965,8 @@ class TFGPT2Model: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFGPT2PreTrainedModel: @@ -974,8 +974,8 @@ class TFGPT2PreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLEDForConditionalGeneration: @@ -983,8 +983,8 @@ class TFLEDForConditionalGeneration: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLEDModel: @@ -992,8 +992,8 @@ class TFLEDModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLEDPreTrainedModel: @@ -1001,8 +1001,8 @@ class TFLEDPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1013,8 +1013,8 @@ class TFLongformerForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLongformerForMultipleChoice: @@ -1022,8 +1022,8 @@ class TFLongformerForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLongformerForQuestionAnswering: @@ -1031,8 +1031,8 @@ class TFLongformerForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLongformerForSequenceClassification: @@ -1040,8 +1040,8 @@ class TFLongformerForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLongformerForTokenClassification: @@ -1049,8 +1049,8 @@ class TFLongformerForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLongformerModel: @@ -1058,8 +1058,8 @@ class TFLongformerModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLongformerSelfAttention: @@ -1085,8 +1085,8 @@ class TFLxmertModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLxmertPreTrainedModel: @@ -1094,8 +1094,8 @@ class TFLxmertPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFLxmertVisualFeatureEncoder: @@ -1108,8 +1108,8 @@ class TFMarianModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMarianMTModel: @@ -1117,8 +1117,8 @@ class TFMarianMTModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMBartForConditionalGeneration: @@ -1126,8 +1126,8 @@ class TFMBartForConditionalGeneration: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMBartModel: @@ -1135,8 +1135,8 @@ class TFMBartModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1147,8 +1147,8 @@ class TFMobileBertForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMobileBertForMultipleChoice: @@ -1156,8 +1156,8 @@ class TFMobileBertForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMobileBertForNextSentencePrediction: @@ -1175,8 +1175,8 @@ class TFMobileBertForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMobileBertForSequenceClassification: @@ -1184,8 +1184,8 @@ class TFMobileBertForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMobileBertForTokenClassification: @@ -1193,8 +1193,8 @@ class TFMobileBertForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMobileBertMainLayer: @@ -1207,8 +1207,8 @@ class TFMobileBertModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMobileBertPreTrainedModel: @@ -1216,8 +1216,8 @@ class TFMobileBertPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_MPNET_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1228,8 +1228,8 @@ class TFMPNetForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMPNetForMultipleChoice: @@ -1237,8 +1237,8 @@ class TFMPNetForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMPNetForQuestionAnswering: @@ -1246,8 +1246,8 @@ class TFMPNetForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMPNetForSequenceClassification: @@ -1255,8 +1255,8 @@ class TFMPNetForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMPNetForTokenClassification: @@ -1264,8 +1264,8 @@ class TFMPNetForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMPNetMainLayer: @@ -1278,8 +1278,8 @@ class TFMPNetModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMPNetPreTrainedModel: @@ -1287,8 +1287,8 @@ class TFMPNetPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMT5EncoderModel: @@ -1296,8 +1296,8 @@ class TFMT5EncoderModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMT5ForConditionalGeneration: @@ -1305,8 +1305,8 @@ class TFMT5ForConditionalGeneration: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFMT5Model: @@ -1314,8 +1314,8 @@ class TFMT5Model: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1326,8 +1326,8 @@ class TFOpenAIGPTDoubleHeadsModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFOpenAIGPTForSequenceClassification: @@ -1335,8 +1335,8 @@ class TFOpenAIGPTForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFOpenAIGPTLMHeadModel: @@ -1344,8 +1344,8 @@ class TFOpenAIGPTLMHeadModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFOpenAIGPTMainLayer: @@ -1358,8 +1358,8 @@ class TFOpenAIGPTModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFOpenAIGPTPreTrainedModel: @@ -1367,8 +1367,8 @@ class TFOpenAIGPTPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFPegasusForConditionalGeneration: @@ -1376,8 +1376,8 @@ class TFPegasusForConditionalGeneration: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFPegasusModel: @@ -1385,8 +1385,8 @@ class TFPegasusModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRagModel: @@ -1394,8 +1394,8 @@ class TFRagModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRagSequenceForGeneration: @@ -1416,8 +1416,8 @@ class TFRobertaForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRobertaForMultipleChoice: @@ -1425,8 +1425,8 @@ class TFRobertaForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRobertaForQuestionAnswering: @@ -1434,8 +1434,8 @@ class TFRobertaForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRobertaForSequenceClassification: @@ -1443,8 +1443,8 @@ class TFRobertaForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRobertaForTokenClassification: @@ -1452,8 +1452,8 @@ class TFRobertaForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRobertaMainLayer: @@ -1466,8 +1466,8 @@ class TFRobertaModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRobertaPreTrainedModel: @@ -1475,8 +1475,8 @@ class TFRobertaPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_ROFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1486,14 +1486,18 @@ class TFRoFormerForCausalLM: def __init__(self, *args, **kwargs): requires_backends(self, ["tf"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) + class TFRoFormerForMaskedLM: def __init__(self, *args, **kwargs): requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRoFormerForMultipleChoice: @@ -1501,8 +1505,8 @@ class TFRoFormerForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRoFormerForQuestionAnswering: @@ -1510,8 +1514,8 @@ class TFRoFormerForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRoFormerForSequenceClassification: @@ -1519,8 +1523,8 @@ class TFRoFormerForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRoFormerForTokenClassification: @@ -1528,8 +1532,8 @@ class TFRoFormerForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRoFormerLayer: @@ -1542,8 +1546,8 @@ class TFRoFormerModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFRoFormerPreTrainedModel: @@ -1551,8 +1555,8 @@ class TFRoFormerPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_T5_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1563,8 +1567,8 @@ class TFT5EncoderModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFT5ForConditionalGeneration: @@ -1572,8 +1576,8 @@ class TFT5ForConditionalGeneration: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFT5Model: @@ -1581,8 +1585,8 @@ class TFT5Model: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFT5PreTrainedModel: @@ -1590,8 +1594,8 @@ class TFT5PreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1607,8 +1611,8 @@ class TFTransfoXLForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFTransfoXLLMHeadModel: @@ -1616,8 +1620,8 @@ class TFTransfoXLLMHeadModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFTransfoXLMainLayer: @@ -1630,8 +1634,8 @@ class TFTransfoXLModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFTransfoXLPreTrainedModel: @@ -1639,8 +1643,8 @@ class TFTransfoXLPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1651,8 +1655,8 @@ class TFXLMForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMForQuestionAnsweringSimple: @@ -1660,8 +1664,8 @@ class TFXLMForQuestionAnsweringSimple: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMForSequenceClassification: @@ -1669,8 +1673,8 @@ class TFXLMForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMForTokenClassification: @@ -1678,8 +1682,8 @@ class TFXLMForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMMainLayer: @@ -1692,8 +1696,8 @@ class TFXLMModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMPreTrainedModel: @@ -1701,8 +1705,8 @@ class TFXLMPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMWithLMHeadModel: @@ -1710,8 +1714,8 @@ class TFXLMWithLMHeadModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1722,8 +1726,8 @@ class TFXLMRobertaForMaskedLM: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMRobertaForMultipleChoice: @@ -1731,8 +1735,8 @@ class TFXLMRobertaForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMRobertaForQuestionAnswering: @@ -1740,8 +1744,8 @@ class TFXLMRobertaForQuestionAnswering: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMRobertaForSequenceClassification: @@ -1749,8 +1753,8 @@ class TFXLMRobertaForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMRobertaForTokenClassification: @@ -1758,8 +1762,8 @@ class TFXLMRobertaForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLMRobertaModel: @@ -1767,8 +1771,8 @@ class TFXLMRobertaModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) TF_XLNET_PRETRAINED_MODEL_ARCHIVE_LIST = None @@ -1779,8 +1783,8 @@ class TFXLNetForMultipleChoice: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLNetForQuestionAnsweringSimple: @@ -1788,8 +1792,8 @@ class TFXLNetForQuestionAnsweringSimple: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLNetForSequenceClassification: @@ -1797,8 +1801,8 @@ class TFXLNetForSequenceClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLNetForTokenClassification: @@ -1806,8 +1810,8 @@ class TFXLNetForTokenClassification: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLNetLMHeadModel: @@ -1815,8 +1819,8 @@ class TFXLNetLMHeadModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLNetMainLayer: @@ -1829,8 +1833,8 @@ class TFXLNetModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class TFXLNetPreTrainedModel: @@ -1838,8 +1842,8 @@ class TFXLNetPreTrainedModel: requires_backends(self, ["tf"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tf"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tf"]) class AdamWeightDecay: diff --git a/src/transformers/utils/dummy_timm_and_vision_objects.py b/src/transformers/utils/dummy_timm_and_vision_objects.py index 33acdf7772..a1da2d14be 100644 --- a/src/transformers/utils/dummy_timm_and_vision_objects.py +++ b/src/transformers/utils/dummy_timm_and_vision_objects.py @@ -9,16 +9,24 @@ class DetrForObjectDetection: def __init__(self, *args, **kwargs): requires_backends(self, ["timm", "vision"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["timm", "vision"]) + class DetrForSegmentation: def __init__(self, *args, **kwargs): requires_backends(self, ["timm", "vision"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["timm", "vision"]) + class DetrModel: def __init__(self, *args, **kwargs): requires_backends(self, ["timm", "vision"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["timm", "vision"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["timm", "vision"]) diff --git a/src/transformers/utils/dummy_timm_objects.py b/src/transformers/utils/dummy_timm_objects.py index bc46f68155..2893931fea 100644 --- a/src/transformers/utils/dummy_timm_objects.py +++ b/src/transformers/utils/dummy_timm_objects.py @@ -9,16 +9,24 @@ class DetrForObjectDetection: def __init__(self, *args, **kwargs): requires_backends(self, ["timm"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["timm"]) + class DetrForSegmentation: def __init__(self, *args, **kwargs): requires_backends(self, ["timm"]) + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["timm"]) + class DetrModel: def __init__(self, *args, **kwargs): requires_backends(self, ["timm"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["timm"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["timm"]) diff --git a/src/transformers/utils/dummy_tokenizers_objects.py b/src/transformers/utils/dummy_tokenizers_objects.py index d707da57fe..ed604c7cea 100644 --- a/src/transformers/utils/dummy_tokenizers_objects.py +++ b/src/transformers/utils/dummy_tokenizers_objects.py @@ -7,8 +7,8 @@ class AlbertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class BartTokenizerFast: @@ -16,8 +16,8 @@ class BartTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class BarthezTokenizerFast: @@ -25,8 +25,8 @@ class BarthezTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class BertTokenizerFast: @@ -34,8 +34,8 @@ class BertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class BigBirdTokenizerFast: @@ -43,8 +43,8 @@ class BigBirdTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class CamembertTokenizerFast: @@ -52,8 +52,8 @@ class CamembertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class CLIPTokenizerFast: @@ -61,8 +61,8 @@ class CLIPTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class ConvBertTokenizerFast: @@ -70,8 +70,8 @@ class ConvBertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class DebertaTokenizerFast: @@ -79,8 +79,8 @@ class DebertaTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class DistilBertTokenizerFast: @@ -88,8 +88,8 @@ class DistilBertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class DPRContextEncoderTokenizerFast: @@ -97,8 +97,8 @@ class DPRContextEncoderTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class DPRQuestionEncoderTokenizerFast: @@ -106,8 +106,8 @@ class DPRQuestionEncoderTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class DPRReaderTokenizerFast: @@ -115,8 +115,8 @@ class DPRReaderTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class ElectraTokenizerFast: @@ -124,8 +124,8 @@ class ElectraTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class FunnelTokenizerFast: @@ -133,8 +133,8 @@ class FunnelTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class GPT2TokenizerFast: @@ -142,8 +142,8 @@ class GPT2TokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class HerbertTokenizerFast: @@ -151,8 +151,8 @@ class HerbertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class LayoutLMTokenizerFast: @@ -160,8 +160,8 @@ class LayoutLMTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class LEDTokenizerFast: @@ -169,8 +169,8 @@ class LEDTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class LongformerTokenizerFast: @@ -178,8 +178,8 @@ class LongformerTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class LxmertTokenizerFast: @@ -187,8 +187,8 @@ class LxmertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class MBart50TokenizerFast: @@ -196,8 +196,8 @@ class MBart50TokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class MBartTokenizerFast: @@ -205,8 +205,8 @@ class MBartTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class MobileBertTokenizerFast: @@ -214,8 +214,8 @@ class MobileBertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class MPNetTokenizerFast: @@ -223,8 +223,8 @@ class MPNetTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class MT5TokenizerFast: @@ -232,8 +232,8 @@ class MT5TokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class OpenAIGPTTokenizerFast: @@ -241,8 +241,8 @@ class OpenAIGPTTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class PegasusTokenizerFast: @@ -250,8 +250,8 @@ class PegasusTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class ReformerTokenizerFast: @@ -259,8 +259,8 @@ class ReformerTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class RetriBertTokenizerFast: @@ -268,8 +268,8 @@ class RetriBertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class RobertaTokenizerFast: @@ -277,8 +277,8 @@ class RobertaTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class RoFormerTokenizerFast: @@ -286,8 +286,8 @@ class RoFormerTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class SqueezeBertTokenizerFast: @@ -295,8 +295,8 @@ class SqueezeBertTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class T5TokenizerFast: @@ -304,8 +304,8 @@ class T5TokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class XLMRobertaTokenizerFast: @@ -313,8 +313,8 @@ class XLMRobertaTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class XLNetTokenizerFast: @@ -322,8 +322,8 @@ class XLNetTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) class PreTrainedTokenizerFast: @@ -331,5 +331,5 @@ class PreTrainedTokenizerFast: requires_backends(self, ["tokenizers"]) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, ["tokenizers"]) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["tokenizers"]) diff --git a/utils/check_dummies.py b/utils/check_dummies.py index fb71ea1536..bd990abac0 100644 --- a/utils/check_dummies.py +++ b/utils/check_dummies.py @@ -39,8 +39,8 @@ class {0}: requires_backends(self, {1}) @classmethod - def from_pretrained(self, *args, **kwargs): - requires_backends(self, {1}) + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, {1}) """ DUMMY_CLASS = """ @@ -103,11 +103,14 @@ def read_init(): def create_dummy_object(name, backend_name): """Create the code for the dummy object corresponding to `name`.""" _pretrained = [ - "Config" "ForCausalLM", + "Config", + "ForCausalLM", "ForConditionalGeneration", "ForMaskedLM", "ForMultipleChoice", + "ForObjectDetection", "ForQuestionAnswering", + "ForSegmentation", "ForSequenceClassification", "ForTokenClassification", "Model",