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,