diff --git a/src/transformers/models/auto/modeling_auto.py b/src/transformers/models/auto/modeling_auto.py index c42839968d1d..1fc7b1775a59 100644 --- a/src/transformers/models/auto/modeling_auto.py +++ b/src/transformers/models/auto/modeling_auto.py @@ -2207,14 +2207,14 @@ class AutoModelForAudioTokenization(_BaseAutoModelClass): class AutoModelWithLMHead(_AutoModelWithLMHead): @classmethod - def from_config(cls, config): + def from_config(cls, config, **kwargs): warnings.warn( "The class `AutoModelWithLMHead` is deprecated and will be removed in a future version. Please use " "`AutoModelForCausalLM` for causal language models, `AutoModelForMaskedLM` for masked language models and " "`AutoModelForSeq2SeqLM` for encoder-decoder models.", FutureWarning, ) - return super().from_config(config) + return super().from_config(config, **kwargs) @classmethod def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs): @@ -2229,13 +2229,13 @@ def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs): class AutoModelForVision2Seq(_AutoModelForVision2Seq): @classmethod - def from_config(cls, config): + def from_config(cls, config, **kwargs): warnings.warn( "The class `AutoModelForVision2Seq` is deprecated and will be removed in v5.0. Please use " "`AutoModelForImageTextToText` instead.", FutureWarning, ) - return super().from_config(config) + return super().from_config(config, **kwargs) @classmethod def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):