diff --git a/src/diffusers/configuration_utils.py b/src/diffusers/configuration_utils.py index 20732581b5eb..f9b652bbc021 100644 --- a/src/diffusers/configuration_utils.py +++ b/src/diffusers/configuration_utils.py @@ -35,6 +35,7 @@ validate_hf_hub_args, ) from requests import HTTPError +from typing_extensions import Self from . import __version__ from .utils import ( @@ -185,7 +186,9 @@ def save_config(self, save_directory: Union[str, os.PathLike], push_to_hub: bool ) @classmethod - def from_config(cls, config: Union[FrozenDict, Dict[str, Any]] = None, return_unused_kwargs=False, **kwargs): + def from_config( + cls, config: Union[FrozenDict, Dict[str, Any]] = None, return_unused_kwargs=False, **kwargs + ) -> Union[Self, Tuple[Self, Dict[str, Any]]]: r""" Instantiate a Python class from a config dictionary. diff --git a/src/diffusers/schedulers/scheduling_utils.py b/src/diffusers/schedulers/scheduling_utils.py index f20224b19009..83f31b72c10b 100644 --- a/src/diffusers/schedulers/scheduling_utils.py +++ b/src/diffusers/schedulers/scheduling_utils.py @@ -19,6 +19,7 @@ import torch from huggingface_hub.utils import validate_hf_hub_args +from typing_extensions import Self from ..utils import BaseOutput, PushToHubMixin @@ -99,7 +100,7 @@ def from_pretrained( subfolder: Optional[str] = None, return_unused_kwargs=False, **kwargs, - ): + ) -> Self: r""" Instantiate a scheduler from a pre-defined JSON configuration file in a local directory or Hub repository.