From 93a2c703643a649efcf6f95c03d97a40d5c4f322 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 07:06:34 -0800 Subject: [PATCH] Update sd_samplers_kdiffusion.py --- modules/sd_samplers_kdiffusion.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py index 5e115486..0190b138 100644 --- a/modules/sd_samplers_kdiffusion.py +++ b/modules/sd_samplers_kdiffusion.py @@ -207,6 +207,11 @@ class KDiffusionSampler(sd_samplers_common.Sampler): [unet_patcher], 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 sigmas = self.get_sigmas(p, steps)