From f2aa2f91dc54506071f7033f0aee442f522be65b Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 16 Apr 2025 12:46:04 +0530 Subject: [PATCH] fix --- src/diffusers/loaders/lora/lora_pipeline.py | 4 ++-- src/diffusers/loaders/unet/unet_loader_utils.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/diffusers/loaders/lora/lora_pipeline.py b/src/diffusers/loaders/lora/lora_pipeline.py index fce66e4393..07ee90891f 100644 --- a/src/diffusers/loaders/lora/lora_pipeline.py +++ b/src/diffusers/loaders/lora/lora_pipeline.py @@ -73,10 +73,10 @@ _MODULE_NAME_TO_ATTRIBUTE_MAP_FLUX = {"x_embedder": "in_channels"} def _maybe_dequantize_weight_for_expanded_lora(model, module): if is_bitsandbytes_available(): - from ..quantizers.bitsandbytes import dequantize_bnb_weight + from ...quantizers.bitsandbytes import dequantize_bnb_weight if is_gguf_available(): - from ..quantizers.gguf.utils import dequantize_gguf_tensor + from ...quantizers.gguf.utils import dequantize_gguf_tensor is_bnb_4bit_quantized = module.weight.__class__.__name__ == "Params4bit" is_gguf_quantized = module.weight.__class__.__name__ == "GGUFParameter" diff --git a/src/diffusers/loaders/unet/unet_loader_utils.py b/src/diffusers/loaders/unet/unet_loader_utils.py index 8f202ed4d4..bfa935d13b 100644 --- a/src/diffusers/loaders/unet/unet_loader_utils.py +++ b/src/diffusers/loaders/unet/unet_loader_utils.py @@ -14,12 +14,12 @@ import copy from typing import TYPE_CHECKING, Dict, List, Union -from ..utils import logging +from ...utils import logging if TYPE_CHECKING: # import here to avoid circular imports - from ..models import UNet2DConditionModel + from ...models import UNet2DConditionModel logger = logging.get_logger(__name__) # pylint: disable=invalid-name