aboutsummaryrefslogtreecommitdiffstats
path: root/modules/processing.py
diff options
context:
space:
mode:
authordrhead <1313496+drhead@users.noreply.github.com>2023-12-02 18:11:00 +0000
committerGitHub <noreply@github.com>2023-12-02 18:11:00 +0000
commit81c4ddf6ebebe6f18338de3b0391da1d8521a525 (patch)
treea7903b0ca8bd6790acb622cdd4973163771ebe18 /modules/processing.py
parent309a606c2fa645b6b8623f96ea56117e685a47fb (diff)
downloadstable-diffusion-webui-gfx803-81c4ddf6ebebe6f18338de3b0391da1d8521a525.tar.gz
stable-diffusion-webui-gfx803-81c4ddf6ebebe6f18338de3b0391da1d8521a525.tar.bz2
stable-diffusion-webui-gfx803-81c4ddf6ebebe6f18338de3b0391da1d8521a525.zip
fix linting
Diffstat (limited to 'modules/processing.py')
-rw-r--r--modules/processing.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/processing.py b/modules/processing.py
index bfa59038..eeccea74 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -884,7 +884,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
if hasattr(p.sd_model, 'alphas_cumprod') and not hasattr(p.sd_model, 'alphas_cumprod_original'):
p.sd_model.alphas_cumprod_original = p.sd_model.alphas_cumprod
-
+
p.sd_model.alphas_cumprod = p.sd_model.alphas_cumprod_original.to(shared.device)
if opts.use_downcasted_alpha_bar: