Fix bug when applying controlnet to negative prompt.

This commit is contained in:
comfyanonymous 2023-03-15 17:58:13 -04:00
parent 494cfe5444
commit d834eed7d0
1 changed files with 2 additions and 2 deletions

View File

@ -691,8 +691,8 @@ def common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive,
if t.shape[0] < noise.shape[0]: if t.shape[0] < noise.shape[0]:
t = torch.cat([t] * noise.shape[0]) t = torch.cat([t] * noise.shape[0])
t = t.to(device) t = t.to(device)
if 'control' in p[1]: if 'control' in n[1]:
control_nets += [p[1]['control']] control_nets += [n[1]['control']]
negative_copy += [[t] + n[1:]] negative_copy += [[t] + n[1:]]
control_net_models = [] control_net_models = []