This commit is contained in:
lllyasviel
2024-01-28 08:19:18 -08:00
parent b066e1b9f6
commit 3293b2b8df
2 changed files with 1 additions and 3 deletions

View File

@@ -72,8 +72,8 @@ def compute_controlnet_weighting(
return control
cond_or_uncond = transformer_options['cond_or_uncond']
cond_or_uncond_size = transformer_options['cond_or_uncond_size']
sigmas = transformer_options['sigmas']
cond_or_uncond_size = int(sigmas.shape[0])
if advanced_sigma_weighting is not None:
advanced_sigma_weighting = advanced_sigma_weighting(sigmas)