fix degradation

This commit is contained in:
lllyasviel 2024-02-09 23:09:01 -08:00
parent ac94f38d9a
commit ce21e7afe4

View File

@ -433,8 +433,8 @@ class ControlNetForForgeOfficial(scripts.Script):
kwargs.update(dict( kwargs.update(dict(
unit=unit, unit=unit,
params=params, params=params,
cond_original=cond.clone() if cond is not None else None, cond_original=cond.clone() if isinstance(cond, torch.Tensor) else cond,
mask_original=mask.clone() if mask is not None else None mask_original=mask.clone() if isinstance(mask, torch.Tensor) else mask,
)) ))
params.model.strength = float(unit.weight) params.model.strength = float(unit.weight)