Fix

huggingface · Nov 7, 2023 · 56ebaee · 56ebaee
1 parent d91ffe9
commit 56ebaee
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/diffusers/models/lora.py b/src/diffusers/models/lora.py
@@ -18,15 +18,14 @@
 import torch.nn.functional as F
 from torch import nn
 
+from ..loaders import PatchedLoraProjection, text_encoder_attn_modules, text_encoder_mlp_modules
 from ..utils import logging
 
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
 
 
 def adjust_lora_scale_text_encoder(text_encoder, lora_scale: float = 1.0):
-    from ..loaders import PatchedLoraProjection, text_encoder_attn_modules, text_encoder_mlp_modules
-
     for _, attn_module in text_encoder_attn_modules(text_encoder):
         if isinstance(attn_module.q_proj, PatchedLoraProjection):
             attn_module.q_proj.lora_scale = lora_scale