Commit ec194b63 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

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

parent f8ff8c06
Loading
Loading
Loading
Loading
+2 −0
Original line number Diff line number Diff line
@@ -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]