feat: SUPIR model support (CORE-17) (#13250)

2026-04-19 06:02:01 +03:00
parent 3086026401
commit b9dedea57d
7 changed files with 680 additions and 11 deletions
@@ -506,6 +506,10 @@ class ModelPatcher:
    def set_model_noise_refiner_patch(self, patch):
        self.set_model_patch(patch, "noise_refiner")

+    def set_model_middle_block_after_patch(self, patch):
+        self.set_model_patch(patch, "middle_block_after_patch")
+
+
    def set_model_rope_options(self, scale_x, shift_x, scale_y, shift_y, scale_t, shift_t, **kwargs):
        rope_options = self.model_options["transformer_options"].get("rope_options", {})
        rope_options["scale_x"] = scale_x