From 153e746b705ab3554292e7a164de6bb787eef1ed Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Fri, 19 Jan 2024 08:43:21 +0000 Subject: [PATCH] clean --- src/diffusers/loaders/single_file.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/diffusers/loaders/single_file.py b/src/diffusers/loaders/single_file.py index 1e4678cfbe..21670385af 100644 --- a/src/diffusers/loaders/single_file.py +++ b/src/diffusers/loaders/single_file.py @@ -18,7 +18,6 @@ from huggingface_hub.utils import validate_hf_hub_args from transformers import AutoFeatureExtractor from ..models.modeling_utils import load_state_dict -from ..pipelines.pipeline_utils import _get_pipeline_class from ..pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker from ..utils import ( is_accelerate_available, @@ -286,6 +285,8 @@ class FromSingleFileMixin: ) return component["controlnet"] + from ..pipelines.pipeline_utils import _get_pipeline_class + pipeline_class = _get_pipeline_class( cls, config=None,