diff --git a/invokeai/backend/model_management/models/lora.py b/invokeai/backend/model_management/models/lora.py index c3f25e6852..88a50fb4fd 100644 --- a/invokeai/backend/model_management/models/lora.py +++ b/invokeai/backend/model_management/models/lora.py @@ -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 diff --git a/invokeai/backend/model_management/util.py b/invokeai/backend/model_management/util.py index 0702224bc7..6d70107c93 100644 --- a/invokeai/backend/model_management/util.py +++ b/invokeai/backend/model_management/util.py @@ -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