Skip to content

Commit ad07796

Browse files
authored
🐛 Add device to variable c (Comfy-Org#5210)
1 parent 1b80895 commit ad07796

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

comfy/controlnet.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -237,6 +237,7 @@ def get_control(self, x_noisy, t, cond, batched_number):
237237
if len(self.extra_concat_orig) > 0:
238238
to_concat = []
239239
for c in self.extra_concat_orig:
240+
c = c.to(self.cond_hint.device)
240241
c = comfy.utils.common_upscale(c, self.cond_hint.shape[3], self.cond_hint.shape[2], self.upscale_algorithm, "center")
241242
to_concat.append(comfy.utils.repeat_to_batch_size(c, self.cond_hint.shape[0]))
242243
self.cond_hint = torch.cat([self.cond_hint] + to_concat, dim=1)

0 commit comments

Comments
 (0)