Skip to content

Commit 3bbba7e

Browse files
committed
Update imports for compatibility with bumped diffusers version.
1 parent b184501 commit 3bbba7e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

invokeai/backend/util/hotfixes.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import diffusers
44
import torch
55
from diffusers.configuration_utils import ConfigMixin, register_to_config
6-
from diffusers.loaders import FromOriginalControlNetMixin
6+
from diffusers.loaders.single_file_model import FromOriginalModelMixin
77
from diffusers.models.attention_processor import AttentionProcessor, AttnProcessor
88
from diffusers.models.controlnet import ControlNetConditioningEmbedding, ControlNetOutput, zero_module
99
from diffusers.models.embeddings import (
@@ -32,7 +32,7 @@
3232
logger = InvokeAILogger.get_logger(__name__)
3333

3434

35-
class ControlNetModel(ModelMixin, ConfigMixin, FromOriginalControlNetMixin):
35+
class ControlNetModel(ModelMixin, ConfigMixin, FromOriginalModelMixin):
3636
"""
3737
A ControlNet model.
3838

0 commit comments

Comments
 (0)