Adapt patch function (hooks)

Acly · Acly · commit 9ba79296fd72 · 2024-12-11T09:58:58.000+01:00
diff --git a/nodes.py b/nodes.py
@@ -67,7 +67,9 @@ def load_fooocus_patch(lora: dict, to_load: dict):
 injected_model_patcher_calculate_weight = False
 
 
-def calculate_weight_patched(patches, weight, key, intermediate_dtype=torch.float32):
+def calculate_weight_patched(
+    patches, weight, key, intermediate_dtype=torch.float32, original_weights=None
+):
     remaining = []
 
     for p in patches:
