lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 11:11:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sd_samplers_timesteps.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 07:39:51 -08:00
parent 01efa9fb7e
commit eacd8cb349
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/sd_samplers_timesteps.py
View File

@@ -56,6 +56,7 @@ class CFGDenoiserTimesteps(CFGDenoiser):
def get_pred_x0(self, x_in, x_out, sigma): def get_pred_x0(self, x_in, x_out, sigma):
ts = sigma.to(dtype=int) ts = sigma.to(dtype=int)
self.alphas = self.alphas.to(ts.device)
a_t = self.alphas[ts][:, None, None, None] a_t = self.alphas[ts][:, None, None, None]
sqrt_one_minus_at = (1 - a_t).sqrt() sqrt_one_minus_at = (1 - a_t).sqrt()