diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-08 22:14:02 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-08 22:14:02 +0300 |
| commit | ec194b637476855ea5918a44a65e85fb587483ab (patch) | |
| tree | 413450ffd1ff342d2e584345da68ec7e1a59a21a /modules | |
| parent | f8ff8c0638997fd0aef217db1505598846f14782 (diff) | |
fix webui not switching back to original model from refiner when batch count is greater than 1
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/processing.py | 2 |
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]
|