github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-19 09:54:24 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update lora_manager.py

Browse Source
This commit is contained in:
Jordan
2023-02-21 02:05:11 -07:00
parent 5529309e73
commit c669336d6b
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/modules/lora_manager.py
View File

@@ -171,9 +171,7 @@ class LoraManager:
return lora
def apply_module_forward(self, module, lora_name):
handle = RemovableHandle(module._forward_hooks)
handle.id = 9000
module._forward_hooks[handle.id] = lora_forward_hook(lora_name)
handle = module.register_module_forward_hook(lora_forward_hook(lora_name))
self.hooks.append(handle)
def apply_lora_model(self, name, mult: float = 1.0):