AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

clean

Browse Source
This commit is contained in:
Dhruv Nair
2024-01-19 08:43:21 +00:00
parent 7395283f23
commit 153e746b70
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/diffusers/loaders/single_file.py
View File

@@ -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,