From 9c1ece89784e36a86b19f371e3b6e60bb630394e Mon Sep 17 00:00:00 2001
From: drhead <1313496+drhead@users.noreply.github.com>
Date: Tue, 20 Feb 2024 19:23:21 -0500
Subject: [PATCH] Protect alphas_cumprod during refiner switchover

---
 modules/sd_samplers_common.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py
index 6bd38e12a..c9578ffe6 100644
--- a/modules/sd_samplers_common.py
+++ b/modules/sd_samplers_common.py
@@ -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()
-- 
GitLab