CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

invert

Browse Source
This commit is contained in:
Haoming 2026-05-26 14:54:16 +08:00
parent b990be9cbe
commit b8dfe84887
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
extensions-builtin/sd_forge_controlllite/scripts/forge_controllllite.py
View File

@ -46,6 +46,9 @@ class ControlLLLiteAnimaPatcher(ControlModelPatcher):
self._lllite_net = self._lllite_net.eval().to(device=device, dtype=dtype)
del self.state_dict
if mask is not None and getattr(process, "inpainting_mask_invert", False):
mask = 1.0 - mask
if (not self._is_inpaint) and (mask is not None):
if inv := self._is_black_on_white(cond):
cond = 1.0 - cond
@ -88,6 +91,8 @@ class ControlLLLitePatcher(ControlModelPatcher):
unet = process.sd_model.forge_objects.unet
if mask is not None:
if getattr(process, "inpainting_mask_invert", False):
mask = 1.0 - mask
cond *= mask
unet = LLLiteLoader.load_lllite(model=unet, state_dict=self.state_dict, cond_image=cond.movedim(1, -1), strength=self.strength, steps=process.steps, start_percent=self.start_percent, end_percent=self.end_percent)