diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index 2e746d8a9e1..0739e92792d 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -487,6 +487,9 @@ def calculate_weight(self, patches, weight, key): weight += ((strength * alpha) * lora_diff).type(weight.dtype) except Exception as e: logging.error("ERROR {} {} {}".format(patch_type, key, e)) + + elif patch_type == "patch_function": + weight = v(p, weight, key) else: logging.warning("patch type not recognized {} {}".format(patch_type, key))