From d49c5294246d59e39f366300d2ae01cd9b50da4f Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 07:41:12 -0800 Subject: [PATCH] Update sd_samplers_timesteps.py --- modules/sd_samplers_timesteps.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/modules/sd_samplers_timesteps.py b/modules/sd_samplers_timesteps.py index 32342252..4d9e6737 100644 --- a/modules/sd_samplers_timesteps.py +++ b/modules/sd_samplers_timesteps.py @@ -154,6 +154,11 @@ class CompVisSampler(sd_samplers_common.Sampler): [unet_patcher], unet_patcher.memory_required([x.shape[0] * 2] + list(x.shape[1:])) + inference_memory) + self.model_wrap.inner_model.betas = self.model_wrap.inner_model.betas.to(unet_patcher.current_device) + self.model_wrap.inner_model.alphas_cumprod = self.model_wrap.inner_model.alphas_cumprod.to(unet_patcher.current_device) + self.model_wrap.inner_model.alphas_cumprod_prev = self.model_wrap.inner_model.alphas_cumprod_prev.to(unet_patcher.current_device) + self.model_wrap.inner_model.logvar = self.model_wrap.inner_model.logvar.to(unet_patcher.current_device) + steps = steps or p.steps timesteps = self.get_timesteps(p, steps)