From c61a0d8c22aca0adb76b5bc1de363cdba1bb27c4 Mon Sep 17 00:00:00 2001 From: hlky Date: Thu, 6 Mar 2025 06:12:06 +0000 Subject: [PATCH] Fix loading OneTrainer Flux LoRA --- src/diffusers/loaders/lora_conversion_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/loaders/lora_conversion_utils.py b/src/diffusers/loaders/lora_conversion_utils.py index e2dd3322fdcb..4be6971755d2 100644 --- a/src/diffusers/loaders/lora_conversion_utils.py +++ b/src/diffusers/loaders/lora_conversion_utils.py @@ -654,6 +654,7 @@ def _convert(original_key, diffusers_key, state_dict, new_state_dict): _convert(k, diffusers_key, state_dict, new_state_dict) + remaining_all_unet = False if state_dict: remaining_all_unet = all(k.startswith("lora_unet_") for k in state_dict) if remaining_all_unet: