Skip to content

GitLab

Unverified Commit 9c7f7fc4 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[ControlNet] fix import in single file loading (#5834) 



fix import
Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
parent 4adad57e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/loaders/single_file.py
View file @ 9c7f7fc4
...@@ -194,8 +194,8 @@ class FromSingleFileMixin: ...@@ -194,8 +194,8 @@ class FromSingleFileMixin:
"StableDiffusionControlNetImg2ImgPipeline", "StableDiffusionControlNetImg2ImgPipeline",
"StableDiffusionControlNetInpaintPipeline", "StableDiffusionControlNetInpaintPipeline",
]: ]:
from .models.controlnet import ControlNetModel from ..models.controlnet import ControlNetModel
from .pipelines.controlnet.multicontrolnet import MultiControlNetModel from ..pipelines.controlnet.multicontrolnet import MultiControlNetModel
# list/tuple or a single instance of ControlNetModel or MultiControlNetModel # list/tuple or a single instance of ControlNetModel or MultiControlNetModel
if not ( if not (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment