lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 10:41:25 +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

1
backend/loader.py
View File

@@ -244,6 +244,7 @@ def split_state_dict(sd, additional_state_dicts: list = None):
guess.clip_target = guess.clip_target(sd) guess.clip_target = guess.clip_target(sd)
guess.model_type = guess.model_type(sd) guess.model_type = guess.model_type(sd)
guess.ztsnr = 'ztsnr' in sd
state_dict = { state_dict = {
guess.unet_target: try_filter_state_dict(sd, guess.unet_key_prefix), guess.unet_target: try_filter_state_dict(sd, guess.unet_key_prefix),

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) sd_models.apply_alpha_schedule_override(p.sd_model, p)
sigmas_backup = None 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 p.extra_generation_params['Noise Schedule'] = opts.sd_noise_schedule
sigmas_backup = p.sd_model.forge_objects.unet.model.predictor.sigmas 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)) p.sd_model.forge_objects.unet.model.predictor.set_sigmas(rescale_zero_terminal_snr_sigmas(p.sd_model.forge_objects.unet.model.predictor.sigmas))