diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py index 0190b138..e3906d1a 100644 --- a/modules/sd_samplers_kdiffusion.py +++ b/modules/sd_samplers_kdiffusion.py @@ -208,8 +208,6 @@ class KDiffusionSampler(sd_samplers_common.Sampler): 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