Fix SD3 trained lora loading and merging

kohya-ss · Nov 7, 2024 · e54462a · e54462a
1 parent 40ed54b
commit e54462a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/networks/lora_sd3.py b/networks/lora_sd3.py
@@ -601,7 +601,7 @@ def merge_to(self, text_encoders, mmdit, weights_sd, dtype=None, device=None):
                 or key.startswith(LoRANetwork.LORA_PREFIX_TEXT_ENCODER_T5)
             ):
                 apply_text_encoder = True
-            elif key.startswith(LoRANetwork.LORA_PREFIX_MMDIT):
+            elif key.startswith(LoRANetwork.LORA_PREFIX_SD3):
                 apply_unet = True
 
         if apply_text_encoder: