Fix potential issue with hydit controlnets.

main
comfyanonymous 6 months ago
parent ec28cd9136
commit 63fafaef45

@ -372,7 +372,7 @@ class HunYuanDiT(nn.Module):
for layer, block in enumerate(self.blocks): for layer, block in enumerate(self.blocks):
if layer > self.depth // 2: if layer > self.depth // 2:
if controls is not None: if controls is not None:
skip = skips.pop() + controls.pop() skip = skips.pop() + controls.pop().to(dtype=x.dtype)
else: else:
skip = skips.pop() skip = skips.pop()
x = block(x, c, text_states, freqs_cis_img, skip) # (N, L, D) x = block(x, c, text_states, freqs_cis_img, skip) # (N, L, D)

Loading…
Cancel
Save