fix webui not switching back to original model from refiner when batch count is greater than 1

1 files changed, 2 insertions, 0 deletions

diff --git a/modules/processing.py b/modules/processing.py
index b635cc74..cf62cdd3 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]