diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index 5e115486..0190b138 100644
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -207,6 +207,11 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
             [unet_patcher],
             unet_patcher.memory_required([x.shape[0] * 2] + list(x.shape[1:])) + inference_memory)
 
+        self.model_wrap.log_sigmas = self.model_wrap.log_sigmas.to(unet_patcher.current_device)
+        self.model_wrap.sigma_min = self.model_wrap.sigma_min.to(unet_patcher.current_device)
+        self.model_wrap.sigma_max = self.model_wrap.sigma_max.to(unet_patcher.current_device)
+        self.model_wrap.sigmas = self.model_wrap.sigmas.to(unet_patcher.current_device)
+
         steps = steps or p.steps
 
         sigmas = self.get_sigmas(p, steps)