From ce21e7afe41da05701431c2a57643fd462486b46 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Fri, 9 Feb 2024 23:09:01 -0800 Subject: [PATCH] fix degradation --- extensions-builtin/sd_forge_controlnet/scripts/controlnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py index dd5e3fa1..e7e228c6 100644 --- a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py +++ b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py @@ -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)