From 0a5ac13b1401e35ccd0cbd4d20185d08df7d9db0 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sun, 4 Feb 2024 01:22:44 -0800 Subject: [PATCH] Update sd_samplers_lcm.py --- modules/sd_samplers_lcm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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))