From 770a56193eeefce7d234fe897e16db7b778128b3 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sat, 1 Apr 2023 09:17:37 +0900 Subject: [PATCH] fix conv2d3x3 is not merged --- networks/merge_lora.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/networks/merge_lora.py b/networks/merge_lora.py index 10174785..2fa8861b 100644 --- a/networks/merge_lora.py +++ b/networks/merge_lora.py @@ -42,7 +42,9 @@ def merge_to_sd_model(text_encoder, unet, models, ratios, merge_dtype): target_replace_modules = lora.LoRANetwork.TEXT_ENCODER_TARGET_REPLACE_MODULE else: prefix = lora.LoRANetwork.LORA_PREFIX_UNET - target_replace_modules = lora.LoRANetwork.UNET_TARGET_REPLACE_MODULE + target_replace_modules = ( + lora.LoRANetwork.UNET_TARGET_REPLACE_MODULE + lora.LoRANetwork.UNET_TARGET_REPLACE_MODULE_CONV2D_3X3 + ) for name, module in root_module.named_modules(): if module.__class__.__name__ in target_replace_modules: