From fc5fabb6291feffb19ad967c3ec080390ae5d435 Mon Sep 17 00:00:00 2001 From: Yousef Rafat <81116377+yousef-rafat@users.noreply.github.com> Date: Mon, 22 Dec 2025 21:16:21 +0200 Subject: [PATCH] . --- comfy/samplers.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/comfy/samplers.py b/comfy/samplers.py index fa09c71af..fa4640842 100644 --- a/comfy/samplers.py +++ b/comfy/samplers.py @@ -984,6 +984,9 @@ class CFGGuider: self.inner_model, self.conds, self.loaded_models = comfy.sampler_helpers.prepare_sampling(self.model_patcher, noise.shape, self.conds, self.model_options) device = self.model_patcher.load_device + if denoise_mask is not None: + denoise_mask = comfy.sampler_helpers.prepare_mask(denoise_mask, noise.shape, device) + noise = noise.to(device) latent_image = latent_image.to(device) sigmas = sigmas.to(device)