Make deep shrink behave like it should.

2023-11-16 15:26:28 -05:00
parent 9f00a18095
commit 7e3fe3ad28
3 changed files with 13 additions and 2 deletions
--- a/comfy/model_patcher.py
+++ b/comfy/model_patcher.py
@@ -99,6 +99,9 @@ class ModelPatcher:
    def set_model_input_block_patch(self, patch):
        self.set_model_patch(patch, "input_block_patch")

+    def set_model_input_block_patch_after_skip(self, patch):
+        self.set_model_patch(patch, "input_block_patch_after_skip")
+
    def set_model_output_block_patch(self, patch):
        self.set_model_patch(patch, "output_block_patch")