Skip to content

Commit 66b0961

Browse files
Fix ControlLora issue with last commit.
1 parent 754597c commit 66b0961

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/controlnet.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ def get_control(self, x_noisy, t, cond, batched_number):
238238
to_concat.append(comfy.utils.repeat_to_batch_size(c, self.cond_hint.shape[0]))
239239
self.cond_hint = torch.cat([self.cond_hint] + to_concat, dim=1)
240240

241-
self.cond_hint = self.cond_hint.to(device=self.load_device, dtype=dtype)
241+
self.cond_hint = self.cond_hint.to(device=x_noisy.device, dtype=dtype)
242242
if x_noisy.shape[0] != self.cond_hint.shape[0]:
243243
self.cond_hint = broadcast_image_to(self.cond_hint, x_noisy.shape[0], batched_number)
244244

0 commit comments

Comments
 (0)