未验证 提交 9c1ece89 编写于 作者: D drhead 提交者: GitHub

Protect alphas_cumprod during refiner switchover

上级 0a271938
......@@ -181,8 +181,12 @@ def apply_refiner(cfg_denoiser):
cfg_denoiser.p.extra_generation_params['Refiner'] = refiner_checkpoint_info.short_title
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
with sd_models.SkipWritingToConfig():
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
devices.torch_gc()
cfg_denoiser.p.setup_conds()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册