ini mask support

2024-02-01 21:21:41 -08:00
parent 07977a1fec
commit 9f6ee2a688
9 changed files with 109 additions and 236 deletions
@@ -63,7 +63,7 @@ class PreprocessorClipVisionForRevision(PreprocessorClipVision):
        self.slider_1 = PreprocessorParameter(
            label="Noise Augmentation", minimum=0.0, maximum=1.0, value=0.0, visible=True)

-    def process_before_every_sampling(self, process, cond, *args, **kwargs):
+    def process_before_every_sampling(self, process, cond, mask, *args, **kwargs):
        unit = kwargs['unit']

        weight = float(unit.weight)
@@ -84,7 +84,8 @@ class PreprocessorClipVisionForRevision(PreprocessorClipVision):
        unet.add_conditioning_modifier(revision_conditioning_modifier, ensure_uniqueness=True)

        process.sd_model.forge_objects.unet = unet
-        return
+
+        return cond, mask


 add_supported_preprocessor(PreprocessorClipVisionForRevision(