github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Relax lora_layers_from_flux_diffusers_grouped_state_dict(...) so that it can work with more LoRA variants (e.g. hada)

Browse Source
This commit is contained in:
Ryan Dick
2025-01-23 20:05:07 +00:00
parent 1054283f5c
commit b8eed2bdcb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/patches/lora_conversions/flux_diffusers_lora_conversion_utils.py
View File

@@ -101,8 +101,8 @@ def lora_layers_from_flux_diffusers_grouped_state_dict(
src_layer_dict = grouped_state_dict.pop(src_key, None)
if src_layer_dict is not None:
values = get_lora_layer_values(src_layer_dict)
assert values["lora_down.weight"].shape[1] == src_weight_shape[1]
assert values["lora_up.weight"].shape[0] == src_weight_shape[0]
# assert values["lora_down.weight"].shape[1] == src_weight_shape[1]
# assert values["lora_up.weight"].shape[0] == src_weight_shape[0]
sub_layers.append(any_lora_layer_from_state_dict(values))
else:
if not allow_missing_keys: