From 5b1dcd15848f6748c6cec978ef962db391c4e4cd Mon Sep 17 00:00:00 2001 From: SahilCarterr <110806554+SahilCarterr@users.noreply.github.com> Date: Wed, 5 Feb 2025 00:29:31 +0530 Subject: [PATCH] [Fix] Type Hint in from_pretrained() to Ensure Correct Type Inference (#10714) * Update pipeline_utils.py Added Self in from_pretrained method so inference will correctly recognize pipeline * Use typing_extensions --------- Co-authored-by: hlky --- src/diffusers/pipelines/pipeline_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index 0c1371c755..c4593b3e69 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -41,6 +41,7 @@ from huggingface_hub.utils import OfflineModeIsEnabled, validate_hf_hub_args from packaging import version from requests.exceptions import HTTPError from tqdm.auto import tqdm +from typing_extensions import Self from .. import __version__ from ..configuration_utils import ConfigMixin @@ -513,7 +514,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): @classmethod @validate_hf_hub_args - def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.PathLike]], **kwargs): + def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.PathLike]], **kwargs) -> Self: r""" Instantiate a PyTorch diffusion pipeline from pretrained pipeline weights.