From d870e3c9a67d291b756e23ff902bb0f3d1123325 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 16 Apr 2025 12:35:09 +0530 Subject: [PATCH] update --- src/diffusers/loaders/lora/lora_pipeline.py | 2 +- src/diffusers/loaders/unet/unet.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/loaders/lora/lora_pipeline.py b/src/diffusers/loaders/lora/lora_pipeline.py index 7a46be4598..fce66e4393 100644 --- a/src/diffusers/loaders/lora/lora_pipeline.py +++ b/src/diffusers/loaders/lora/lora_pipeline.py @@ -18,7 +18,7 @@ from typing import Callable, Dict, List, Optional, Union import torch from huggingface_hub.utils import validate_hf_hub_args -from ..utils import ( +from ...utils import ( USE_PEFT_BACKEND, deprecate, get_submodule_by_name, diff --git a/src/diffusers/loaders/unet/unet.py b/src/diffusers/loaders/unet/unet.py index fd86d26be1..a632ceac21 100644 --- a/src/diffusers/loaders/unet/unet.py +++ b/src/diffusers/loaders/unet/unet.py @@ -43,7 +43,7 @@ from ...utils import ( is_torch_version, logging, ) -from ..lora import _func_optionally_disable_offloading +from ..lora.lora_base import _func_optionally_disable_offloading from ..lora.lora_pipeline import LORA_WEIGHT_NAME, LORA_WEIGHT_NAME_SAFE, TEXT_ENCODER_NAME, UNET_NAME from ..utils import AttnProcsLayers