From c669336d6b3958a4befe92e682d72a1e15b8f86d Mon Sep 17 00:00:00 2001 From: Jordan Date: Tue, 21 Feb 2023 02:05:11 -0700 Subject: [PATCH] Update lora_manager.py --- ldm/modules/lora_manager.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/ldm/modules/lora_manager.py b/ldm/modules/lora_manager.py index 0071bf5650..1c4906331f 100644 --- a/ldm/modules/lora_manager.py +++ b/ldm/modules/lora_manager.py @@ -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):