Skip to content

S
Stable Diffusion Webui
Commit 309a606c authored by drhead's avatar drhead Committed by GitHub
Browse files
Options

ensure that original alpha bar always exists

parent 668ae34e
Show whitespace changes
Inline Side-by-side
Showing with 11 additions and 9 deletions
modules/processing.py
View file @ 309a606c
...@@ -882,7 +882,9 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: ...@@ -882,7 +882,9 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
alphas_bar[-1] = 4.8973451890853435e-08 alphas_bar[-1] = 4.8973451890853435e-08
return alphas_bar return alphas_bar
if hasattr(p.sd_model, 'alphas_cumprod') and hasattr(p.sd_model, 'alphas_cumprod_original'): 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) p.sd_model.alphas_cumprod = p.sd_model.alphas_cumprod_original.to(shared.device)
if opts.use_downcasted_alpha_bar: if opts.use_downcasted_alpha_bar:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment