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

try solve #71

Browse Source
This commit is contained in:
lllyasviel
2024-02-06 17:46:23 -08:00
parent 6301a6660e
commit e579fab4d0
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/sd_samplers_cfg_denoiser.py
View File

@@ -154,6 +154,9 @@ class CFGDenoiser(torch.nn.Module):
if state.interrupted or state.skipped: if state.interrupted or state.skipped:
raise sd_samplers_common.InterruptedException raise sd_samplers_common.InterruptedException
original_x_device = x.device
original_x_dtype = x.dtype
if self.classic_ddim_eps_estimation: if self.classic_ddim_eps_estimation:
acd = self.inner_model.inner_model.alphas_cumprod acd = self.inner_model.inner_model.alphas_cumprod
fake_sigmas = ((1 - acd) / acd) ** 0.5 fake_sigmas = ((1 - acd) / acd) ** 0.5
@@ -195,5 +198,5 @@ class CFGDenoiser(torch.nn.Module):
eps = (x - denoised) / sigma eps = (x - denoised) / sigma
return eps return eps
return denoised return denoised.to(device=original_x_device, dtype=original_x_dtype)