Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change layer detection keys according to LyCORIS repository

Browse Source
This commit is contained in:
Sergey Borisov 2024-07-25 02:10:47 +03:00
parent 653f63ae71
commit 46c632e7cc
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/backend/lora.py
View File

@ -490,15 +490,15 @@ class LoRAModelRaw(RawModel): # (torch.nn.Module):
for layer_key, values in state_dict.items(): for layer_key, values in state_dict.items():
# lora and locon # lora and locon
if "lora_down.weight" in values: if "lora_up.weight" in values:
layer: AnyLoRALayer = LoRALayer(layer_key, values) layer: AnyLoRALayer = LoRALayer(layer_key, values)
# loha # loha
elif "hada_w1_b" in values: elif "hada_w1_a" in values:
layer = LoHALayer(layer_key, values) layer = LoHALayer(layer_key, values)
# lokr # lokr
elif "lokr_w1_b" in values or "lokr_w1" in values: elif "lokr_w1" in values or "lokr_w1_a" in values:
layer = LoKRLayer(layer_key, values) layer = LoKRLayer(layer_key, values)
# diff # diff
@ -506,7 +506,7 @@ class LoRAModelRaw(RawModel): # (torch.nn.Module):
layer = FullLayer(layer_key, values) layer = FullLayer(layer_key, values)
# ia3 # ia3
elif "weight" in values and "on_input" in values: elif "on_input" in values:
layer = IA3Layer(layer_key, values) layer = IA3Layer(layer_key, values)
else: else: