From ec194b637476855ea5918a44a65e85fb587483ab Mon Sep 17 00:00:00 2001
From: AUTOMATIC1111 <16777216c@gmail.com>
Date: Tue, 8 Aug 2023 22:14:02 +0300
Subject: [PATCH] fix webui not switching back to original model from refiner
 when batch count is greater than 1

---
 modules/processing.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/processing.py b/modules/processing.py
index b635cc745..cf62cdd33 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -766,6 +766,8 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
             if state.interrupted:
                 break
 
+            sd_models.reload_model_weights()  # model can be changed for example by refiner
+
             p.prompts = p.all_prompts[n * p.batch_size:(n + 1) * p.batch_size]
             p.negative_prompts = p.all_negative_prompts[n * p.batch_size:(n + 1) * p.batch_size]
             p.seeds = p.all_seeds[n * p.batch_size:(n + 1) * p.batch_size]