diff --git a/comfy/ldm/hydit/models.py b/comfy/ldm/hydit/models.py index f3afaad3..44e806cb 100644 --- a/comfy/ldm/hydit/models.py +++ b/comfy/ldm/hydit/models.py @@ -372,7 +372,7 @@ class HunYuanDiT(nn.Module): for layer, block in enumerate(self.blocks): if layer > self.depth // 2: if controls is not None: - skip = skips.pop() + controls.pop() + skip = skips.pop() + controls.pop().to(dtype=x.dtype) else: skip = skips.pop() x = block(x, c, text_states, freqs_cis_img, skip) # (N, L, D)