fix multidiffusion integrated with controlnet (#2199)

This commit is contained in:
licyk
2024-10-28 20:15:37 +08:00
committed by GitHub
parent 878c056576
commit b592142f3b

View File

@@ -578,9 +578,8 @@ class MixtureOfDiffusers(AbstractDiffusion):
# controlnet
# self.switch_controlnet_tensors(batch_id, N, len(bboxes), is_denoise=True)
if 'control' in c_in:
control = c_in['control']
self.process_controlnet(x_tile.shape, x_tile.dtype, c_in, cond_or_uncond, bboxes, N, batch_id)
c_tile['control'] = control.get_control(x_tile, t_tile, c_tile, len(cond_or_uncond))
c_tile['control'] = c_in['control_model'].get_control(x_tile, t_tile, c_tile, len(cond_or_uncond))
# stablesr
# self.switch_stablesr_tensors(batch_id)