Update sd_samplers_lcm.py
This commit is contained in:
parent
e007d405d5
commit
0a5ac13b14
@ -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))
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user