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

Format by black

Browse Source
This commit is contained in:
Sergey Borisov 2023-08-11 03:20:56 +03:00
parent 56023bc725
commit c0c2ab3dcf
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/backend/model_management/models/lora.py
View File

@ -393,6 +393,7 @@ class FullLayer(LoRALayerBase):
self.weight = self.weight.to(device=device, dtype=dtype)
class IA3Layer(LoRALayerBase):
# weight: torch.Tensor
# on_input: torch.Tensor

2
invokeai/backend/model_management/util.py
View File

@ -13,7 +13,7 @@ def lora_token_vector_length(checkpoint: dict) -> int:
lora_token_vector_length = None
if "." not in key:
return lora_token_vector_length # wrong key format
return lora_token_vector_length # wrong key format
model_key, lora_key = key.split(".", 1)
# check lora/locon