Skip to content

Commit 3129606

Browse files
Kevin Turnerpsychedelicious
authored andcommitted
fix: move AI Toolkit to the bottom of the detection list
to avoid disrupting already-working LoRA
1 parent 50cf285 commit 3129606

File tree

1 file changed

+2
-2
lines changed
  • invokeai/backend/patches/lora_conversions

1 file changed

+2
-2
lines changed

invokeai/backend/patches/lora_conversions/formats.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,6 @@
1515

1616

1717
def flux_format_from_state_dict(state_dict: dict, metadata: dict | None = None) -> FluxLoRAFormat | None:
18-
if is_state_dict_likely_in_flux_aitoolkit_format(state_dict, metadata):
19-
return FluxLoRAFormat.AIToolkit
2018
if is_state_dict_likely_in_flux_kohya_format(state_dict):
2119
return FluxLoRAFormat.Kohya
2220
elif is_state_dict_likely_in_flux_onetrainer_format(state_dict):
@@ -25,5 +23,7 @@ def flux_format_from_state_dict(state_dict: dict, metadata: dict | None = None)
2523
return FluxLoRAFormat.Diffusers
2624
elif is_state_dict_likely_flux_control(state_dict):
2725
return FluxLoRAFormat.Control
26+
elif is_state_dict_likely_in_flux_aitoolkit_format(state_dict, metadata):
27+
return FluxLoRAFormat.AIToolkit
2828
else:
2929
return None

0 commit comments

Comments
 (0)