remove debug print

This commit is contained in:
drhead 2023-11-29 22:48:31 -05:00 committed by GitHub
parent de79597ab9
commit 668ae34e21
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -890,7 +890,6 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
p.sd_model.alphas_cumprod = p.sd_model.alphas_cumprod.half().to(shared.device) p.sd_model.alphas_cumprod = p.sd_model.alphas_cumprod.half().to(shared.device)
if opts.sd_noise_schedule == "Zero Terminal SNR": if opts.sd_noise_schedule == "Zero Terminal SNR":
p.extra_generation_params['Noise Schedule'] = opts.sd_noise_schedule p.extra_generation_params['Noise Schedule'] = opts.sd_noise_schedule
print("rescaling noise schedule for zero snr")
p.sd_model.alphas_cumprod = rescale_zero_terminal_snr_abar(p.sd_model.alphas_cumprod).to(shared.device) p.sd_model.alphas_cumprod = rescale_zero_terminal_snr_abar(p.sd_model.alphas_cumprod).to(shared.device)
with devices.without_autocast() if devices.unet_needs_upcast else devices.autocast(): with devices.without_autocast() if devices.unet_needs_upcast else devices.autocast():