Add a way to add patches to the input block.

2023-11-14 00:08:12 -05:00
parent 8509bd58b4
commit 94cc718e9c
2 changed files with 8 additions and 0 deletions
--- a/comfy/model_patcher.py
+++ b/comfy/model_patcher.py
@@ -96,6 +96,9 @@ class ModelPatcher:
    def set_model_attn2_output_patch(self, patch):
        self.set_model_patch(patch, "attn2_output_patch")

+    def set_model_input_block_patch(self, patch):
+        self.set_model_patch(patch, "input_block_patch")
+
    def set_model_output_block_patch(self, patch):
        self.set_model_patch(patch, "output_block_patch")