lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix mask can be none

Browse Source
This commit is contained in:
lllyasviel
2024-02-09 18:25:44 -08:00
parent fb2e271668
commit 4ec1015162
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
View File

@@ -417,7 +417,12 @@ class ControlNetForForgeOfficial(scripts.Script):
cond = params.control_cond
mask = params.control_mask
kwargs.update(dict(unit=unit, params=params, cond_original=cond.clone(), mask_original=mask.clone()))
kwargs.update(dict(
unit=unit,
params=params,
cond_original=cond.clone() if cond is not None else None,
mask_original=mask.clone() if mask is not None else None
))
params.model.strength = float(unit.weight)
params.model.start_percent = float(unit.guidance_start)