Skip to content

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

dont need to preserve alphas_cumprod_original

parent 9c1ece89
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
modules/sd_samplers_common.py
View file @ 648f6a8e
...@@ -181,11 +181,9 @@ def apply_refiner(cfg_denoiser): ...@@ -181,11 +181,9 @@ def apply_refiner(cfg_denoiser):
cfg_denoiser.p.extra_generation_params['Refiner'] = refiner_checkpoint_info.short_title cfg_denoiser.p.extra_generation_params['Refiner'] = refiner_checkpoint_info.short_title
cfg_denoiser.p.extra_generation_params['Refiner switch at'] = refiner_switch_at cfg_denoiser.p.extra_generation_params['Refiner switch at'] = refiner_switch_at
alphas_cumprod_original = cfg_denoiser.p.sd_model.alphas_cumprod_original
alphas_cumprod = cfg_denoiser.p.sd_model.alphas_cumprod alphas_cumprod = cfg_denoiser.p.sd_model.alphas_cumprod
with sd_models.SkipWritingToConfig(): with sd_models.SkipWritingToConfig():
sd_models.reload_model_weights(info=refiner_checkpoint_info) sd_models.reload_model_weights(info=refiner_checkpoint_info)
cfg_denoiser.p.sd_model.alphas_cumprod_original = alphas_cumprod_original
cfg_denoiser.p.sd_model.alphas_cumprod = alphas_cumprod cfg_denoiser.p.sd_model.alphas_cumprod = alphas_cumprod
devices.torch_gc() devices.torch_gc()
......
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