From eacd8cb34948370aab37a9ec44638524a85b64a6 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 07:39:51 -0800 Subject: [PATCH] Update sd_samplers_timesteps.py --- modules/sd_samplers_timesteps.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/sd_samplers_timesteps.py b/modules/sd_samplers_timesteps.py index 7dd1ade1..4d9e6737 100644 --- a/modules/sd_samplers_timesteps.py +++ b/modules/sd_samplers_timesteps.py @@ -56,6 +56,7 @@ class CFGDenoiserTimesteps(CFGDenoiser): def get_pred_x0(self, x_in, x_out, sigma): ts = sigma.to(dtype=int) + self.alphas = self.alphas.to(ts.device) a_t = self.alphas[ts][:, None, None, None] sqrt_one_minus_at = (1 - a_t).sqrt()