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

fix: Other lora missing type

Browse Source
This commit is contained in:
blessedcoolant 2023-07-24 18:58:24 +12:00
parent 0beec08d38
commit 0cf7a10c5c
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_management/lora.py
View File

@ -474,7 +474,7 @@ class ModelPatcher:
@staticmethod @staticmethod
def _lora_forward_hook( def _lora_forward_hook(
applied_loras: List[Tuple[LoraModel, float]], applied_loras: List[Tuple[LoRAModel, float]],
layer_name: str, layer_name: str,
): ):
@ -519,7 +519,7 @@ class ModelPatcher:
def apply_lora( def apply_lora(
cls, cls,
model: torch.nn.Module, model: torch.nn.Module,
loras: List[Tuple[LoraModel, float]], loras: List[Tuple[LoRAModel, float]],
prefix: str, prefix: str,
): ):
original_weights = dict() original_weights = dict()