Only apply ztSNR related code if alphas_cumprod exists

This commit is contained in:
catboxanon 2023-11-29 18:33:32 -05:00
parent ffa7f8201d
commit de79597ab9

View File

@ -882,6 +882,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
alphas_bar[-1] = 4.8973451890853435e-08
return alphas_bar
if hasattr(p.sd_model, 'alphas_cumprod') and hasattr(p.sd_model, 'alphas_cumprod_original'):
p.sd_model.alphas_cumprod = p.sd_model.alphas_cumprod_original.to(shared.device)
if opts.use_downcasted_alpha_bar: