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

fix(mm): lora state dict loading in model id

Browse Source
This commit is contained in:
psychedelicious
2025-10-10 17:15:25 +11:00
parent 5a006b96fa
commit 020ac1aaa8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_manager/configs/lora.py
View File

@@ -55,7 +55,7 @@ def _get_flux_lora_format(mod: ModelOnDisk) -> FluxLoRAFormat | None:
# TODO(psyche): Moving this import to the function to avoid circular imports. Refactor later.
from invokeai.backend.patches.lora_conversions.formats import flux_format_from_state_dict
state_dict = mod.load_state_dict(mod.path)
state_dict = mod.load_state_dict()
value = flux_format_from_state_dict(state_dict, mod.metadata())
return value