This commit is contained in:
Jordan
2023-02-21 02:08:02 -07:00
parent c669336d6b
commit 24d92979db

View File

@@ -171,7 +171,7 @@ class LoraManager:
return lora
def apply_module_forward(self, module, lora_name):
handle = module.register_module_forward_hook(lora_forward_hook(lora_name))
handle = module.register_forward_hook(lora_forward_hook(lora_name))
self.hooks.append(handle)
def apply_lora_model(self, name, mult: float = 1.0):