From 8b0bc8b1259d7bd6f797dabd5d55a93a84c624aa Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tolga=20Cang=C3=B6z?= Date: Thu, 18 Apr 2024 19:23:55 +0300 Subject: [PATCH] Fix imports --- src/diffusers/pipelines/stable_diffusion/__init__.py | 1 - src/diffusers/schedulers/deprecated/__init__.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/stable_diffusion/__init__.py b/src/diffusers/pipelines/stable_diffusion/__init__.py index 0eda32d333b9..8ce08aec66ca 100644 --- a/src/diffusers/pipelines/stable_diffusion/__init__.py +++ b/src/diffusers/pipelines/stable_diffusion/__init__.py @@ -113,7 +113,6 @@ from .pipeline_stable_diffusion import ( StableDiffusionPipeline, StableDiffusionPipelineOutput, - StableDiffusionSafetyChecker, ) from .pipeline_stable_diffusion_img2img import StableDiffusionImg2ImgPipeline from .pipeline_stable_diffusion_inpaint import StableDiffusionInpaintPipeline diff --git a/src/diffusers/schedulers/deprecated/__init__.py b/src/diffusers/schedulers/deprecated/__init__.py index 786707f45206..479cf9bd568b 100644 --- a/src/diffusers/schedulers/deprecated/__init__.py +++ b/src/diffusers/schedulers/deprecated/__init__.py @@ -30,7 +30,7 @@ raise OptionalDependencyNotAvailable() except OptionalDependencyNotAvailable: - from ..utils.dummy_pt_objects import * # noqa F403 + from ...utils.dummy_pt_objects import * # noqa F403 else: from .scheduling_karras_ve import KarrasVeScheduler from .scheduling_sde_vp import ScoreSdeVpScheduler