soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

fix degradation

Browse Source
This commit is contained in:
lllyasviel 2024-02-09 23:09:01 -08:00
parent ac94f38d9a
commit ce21e7afe4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -433,8 +433,8 @@ class ControlNetForForgeOfficial(scripts.Script):
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
cond_original=cond.clone() if isinstance(cond, torch.Tensor) else cond,
mask_original=mask.clone() if isinstance(mask, torch.Tensor) else mask,
))
params.model.strength = float(unit.weight)