From ea0ce4bfabfeb06b63857f027f5e2dc5030fb6df Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 16 Apr 2025 12:50:09 +0530 Subject: [PATCH] fixes --- src/diffusers/models/unets/unet_2d_condition.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/diffusers/models/unets/unet_2d_condition.py b/src/diffusers/models/unets/unet_2d_condition.py index 5674d8ba26..a35aa1a5b3 100644 --- a/src/diffusers/models/unets/unet_2d_condition.py +++ b/src/diffusers/models/unets/unet_2d_condition.py @@ -19,8 +19,7 @@ import torch.nn as nn import torch.utils.checkpoint from ...configuration_utils import ConfigMixin, register_to_config -from ...loaders import PeftAdapterMixin, UNet2DConditionLoadersMixin -from ...loaders.single_file_model import FromOriginalModelMixin +from ...loaders import FromOriginalModelMixin, PeftAdapterMixin, UNet2DConditionLoadersMixin from ...utils import USE_PEFT_BACKEND, BaseOutput, deprecate, logging, scale_lora_layers, unscale_lora_layers from ..activations import get_activation from ..attention_processor import (