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