From 2f7e387246c5d4341ecef5b995cd7e312bb88688 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Wed, 15 Nov 2023 01:20:36 +0300 Subject: [PATCH] Fix missing import --- extensions-builtin/Lora/networks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/Lora/networks.py b/extensions-builtin/Lora/networks.py index 83df1699d..34b8af088 100644 --- a/extensions-builtin/Lora/networks.py +++ b/extensions-builtin/Lora/networks.py @@ -14,7 +14,7 @@ import network_norm import lora_convert import torch import diffusers.models.lora -from modules import shared, devices, sd_models, errors, scripts, sd_hijack +from modules import shared, devices, sd_models, sd_models_compile, errors, scripts, sd_hijack debug = os.environ.get('SD_LORA_DEBUG', None) @@ -193,7 +193,7 @@ def load_networks(names, te_multipliers=None, unet_multipliers=None, dyn_dims=No if recompile_model: shared.log.info("LoRA recompiling model") - sd_models.compile_diffusers(shared.sd_model) + sd_models_compile.compile_diffusers(shared.sd_model) def network_restore_weights_from_backup(self: Union[torch.nn.Conv2d, torch.nn.Linear, torch.nn.GroupNorm, torch.nn.LayerNorm, torch.nn.MultiheadAttention, diffusers.models.lora.LoRACompatibleLinear, diffusers.models.lora.LoRACompatibleConv]):