diff --git a/modules/sd_samplers_lcm.py b/modules/sd_samplers_lcm.py
index 45df4db9..b1c1e475 100644
--- a/modules/sd_samplers_lcm.py
+++ b/modules/sd_samplers_lcm.py
@@ -27,7 +27,7 @@ class LCMCompVisDenoiser(DiscreteEpsDDPMDenoiser):
         start = self.sigma_to_t(self.sigma_max)
         end = self.sigma_to_t(self.sigma_min)
 
-        t = torch.linspace(start, end, n, device=shared.sd_model.device)
+        t = torch.linspace(start, end, n, device=shared.sd_model.forge_objects.unet.current_device)
 
         return sampling.append_zero(self.t_to_sigma(t))