Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "b13931af4552916867a2da9324b45fe761960921"
Unverified Commit b1502763 authored by hlky's avatar hlky Committed by GitHub
Browse files

Fix loading OneTrainer Flux LoRA (#10978) 


Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 6e2a93de
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/loaders/lora_conversion_utils.py
View file @ b1502763
...@@ -654,6 +654,7 @@ def _convert_kohya_flux_lora_to_diffusers(state_dict): ...@@ -654,6 +654,7 @@ def _convert_kohya_flux_lora_to_diffusers(state_dict):
_convert(k, diffusers_key, state_dict, new_state_dict) _convert(k, diffusers_key, state_dict, new_state_dict)
remaining_all_unet = False
if state_dict: if state_dict:
remaining_all_unet = all(k.startswith("lora_unet_") for k in state_dict) remaining_all_unet = all(k.startswith("lora_unet_") for k in state_dict)
if remaining_all_unet: if remaining_all_unet:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment