~/Projects/stable-diffusion-webui
git clone https://code.lsong.org/stable-diffusion-webui
Commit
- Commit
- ec194b637476855ea5918a44a65e85fb587483ab
- Author
- AUTOMATIC1111 <[email protected]>
- Date
- 2023-08-08 22:14:02 +0300 +0300
- Diffstat
modules/processing.py | 2 ++
fix webui not switching back to original model from refiner when batch count is greater than 1
diff --git a/modules/processing.py b/modules/processing.py
index b635cc74599bb05c1d113fc8372347fd4b294dd3..cf62cdd33b5c3311d11b131dac90d4e6a050c224 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -766,6 +766,8 @@
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]
