diff --git a/modules/sd_samplers_timesteps.py b/modules/sd_samplers_timesteps.py
index 32342252..4d9e6737 100644
--- a/modules/sd_samplers_timesteps.py
+++ b/modules/sd_samplers_timesteps.py
@@ -154,6 +154,11 @@ class CompVisSampler(sd_samplers_common.Sampler):
             [unet_patcher],
             unet_patcher.memory_required([x.shape[0] * 2] + list(x.shape[1:])) + inference_memory)
 
+        self.model_wrap.inner_model.betas = self.model_wrap.inner_model.betas.to(unet_patcher.current_device)
+        self.model_wrap.inner_model.alphas_cumprod = self.model_wrap.inner_model.alphas_cumprod.to(unet_patcher.current_device)
+        self.model_wrap.inner_model.alphas_cumprod_prev = self.model_wrap.inner_model.alphas_cumprod_prev.to(unet_patcher.current_device)
+        self.model_wrap.inner_model.logvar = self.model_wrap.inner_model.logvar.to(unet_patcher.current_device)
+
         steps = steps or p.steps
         timesteps = self.get_timesteps(p, steps)