lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-03 22:58:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Automatically enable ztSNR when applicable (#2122)

Browse Source
This commit is contained in:
catboxanon
2024-10-19 20:33:34 -04:00
committed by GitHub
parent a27a0c6936
commit edc46380cc
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/processing.py
View File

@@ -974,7 +974,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
sd_models.apply_alpha_schedule_override(p.sd_model, p)
sigmas_backup = None
if opts.sd_noise_schedule == "Zero Terminal SNR" and p is not None:
if (opts.sd_noise_schedule == "Zero Terminal SNR" or (hasattr(p.sd_model.model_config, 'ztsnr') and p.sd_model.model_config.ztsnr)) and p is not None:
p.extra_generation_params['Noise Schedule'] = opts.sd_noise_schedule
sigmas_backup = p.sd_model.forge_objects.unet.model.predictor.sigmas
p.sd_model.forge_objects.unet.model.predictor.set_sigmas(rescale_zero_terminal_snr_sigmas(p.sd_model.forge_objects.unet.model.predictor.sigmas))