From 66b0961a468e1ac120f9746cf9091fedff73c0a1 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Wed, 23 Oct 2024 17:02:19 -0400 Subject: [PATCH] Fix ControlLora issue with last commit. --- comfy/controlnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/controlnet.py b/comfy/controlnet.py index 2c0305b4..d2744e42 100644 --- a/comfy/controlnet.py +++ b/comfy/controlnet.py @@ -238,7 +238,7 @@ class ControlNet(ControlBase): to_concat.append(comfy.utils.repeat_to_batch_size(c, self.cond_hint.shape[0])) self.cond_hint = torch.cat([self.cond_hint] + to_concat, dim=1) - self.cond_hint = self.cond_hint.to(device=self.load_device, dtype=dtype) + self.cond_hint = self.cond_hint.to(device=x_noisy.device, dtype=dtype) if x_noisy.shape[0] != self.cond_hint.shape[0]: self.cond_hint = broadcast_image_to(self.cond_hint, x_noisy.shape[0], batched_number)