masked cnet

2024-02-01 21:48:58 -08:00
parent 597fe3e049
commit 4b27699e65
2 changed files with 7 additions and 1 deletions
@@ -352,6 +352,9 @@ class ControlNetForForgeOfficial(scripts.Script):
            params.model.negative_advanced_weighting = soft_weighting.copy()

        cond, mask = params.preprocessor.process_before_every_sampling(p, cond, mask, *args, **kwargs)
+
+        params.model.advanced_mask_weighting = mask
+
        params.model.process_before_every_sampling(p, cond, mask, *args, **kwargs)

        logger.info(f"ControlNet Method {params.preprocessor.name} patched.")