DiffusionAi/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2025-04-04 22:43:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Caching

Browse Source
This commit is contained in:
Billy 2025-03-27 17:55:09 +11:00
parent 323d409fb6
commit 0d75c99476
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/model_manager/config.py
View File

@ -377,12 +377,12 @@ class LoRAConfigBase(ABC, BaseModel):
@classmethod
def flux_lora_format(cls, mod: ModelOnDisk):
from invokeai.backend.patches.lora_conversions.formats import flux_format_from_state_dict
key = "FLUX_LORA_FORMAT"
if key in mod.cache:
return mod.cache[key]
from invokeai.backend.patches.lora_conversions.formats import flux_format_from_state_dict
sd = mod.load_state_dict(mod.path)
value = flux_format_from_state_dict(sd)
mod.cache[key] = value