Skip to content

Commit 63fafae

Browse files
Fix potential issue with hydit controlnets.
1 parent ec28cd9 commit 63fafae

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy/ldm/hydit/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -372,7 +372,7 @@ def forward(self,
372372
for layer, block in enumerate(self.blocks):
373373
if layer > self.depth // 2:
374374
if controls is not None:
375-
skip = skips.pop() + controls.pop()
375+
skip = skips.pop() + controls.pop().to(dtype=x.dtype)
376376
else:
377377
skip = skips.pop()
378378
x = block(x, c, text_states, freqs_cis_img, skip) # (N, L, D)

0 commit comments

Comments
 (0)