From c1d5c24bc7db673b7256bce39e7a2ece033ff033 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sun, 23 Jul 2023 15:01:41 +0900 Subject: [PATCH] fix LoRA with text encoder can't merge closes #660 --- networks/sdxl_merge_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/networks/sdxl_merge_lora.py b/networks/sdxl_merge_lora.py index d75da7d7..a91b62d8 100644 --- a/networks/sdxl_merge_lora.py +++ b/networks/sdxl_merge_lora.py @@ -204,7 +204,7 @@ def merge(args): ckpt_info, ) = sdxl_model_util.load_models_from_sdxl_checkpoint(sdxl_model_util.MODEL_VERSION_SDXL_BASE_V0_9, args.sd_model, "cpu") - merge_to_sd_model(text_model2, text_model2, unet, args.models, args.ratios, merge_dtype) + merge_to_sd_model(text_model1, text_model2, unet, args.models, args.ratios, merge_dtype) print(f"saving SD model to: {args.save_to}") sdxl_model_util.save_stable_diffusion_checkpoint(