From 781a91432a84dc02dbdad7ae1ac552d51c563d13 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Tue, 9 Sep 2025 14:01:50 +0530 Subject: [PATCH] allow from_single_file to be called from automodel. --- src/diffusers/models/auto_model.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/diffusers/models/auto_model.py b/src/diffusers/models/auto_model.py index bfe386f1f6..67c6ae1c29 100644 --- a/src/diffusers/models/auto_model.py +++ b/src/diffusers/models/auto_model.py @@ -16,8 +16,10 @@ import os from typing import Optional, Union from huggingface_hub.utils import validate_hf_hub_args +from typing_extensions import Self from ..configuration_utils import ConfigMixin +from ..loaders import FromOriginalModelMixin from ..utils import logging @@ -204,3 +206,8 @@ class AutoModel(ConfigMixin): kwargs = {**load_config_kwargs, **kwargs} return model_cls.from_pretrained(pretrained_model_or_path, **kwargs) + + @classmethod + @validate_hf_hub_args + def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] = None, **kwargs) -> Self: + return FromOriginalModelMixin.from_single_file(pretrained_model_link_or_path_or_dict, **kwargs)