model load exception handling

2026-09-17 08:19:11 +02:00 · 2023-07-13 10:45:58 -04:00
parent 186bd236f5
commit fdb76ddf12
4 changed files with 8 additions and 2 deletions
@@ -761,6 +761,8 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
                        model=shared.sd_refiner,
                        prompt=[p.refiner_prompt] if len(p.refiner_prompt) > 0 else prompts,
                        negative_prompt=[p.refiner_negative] if len(p.refiner_negative) > 0 else negative_prompts,
+                        num_inference_steps=p.refiner_steps,
+                        denoising_start=p.refiner_denoise,
                        image=output.images[0],
                        output_type='np'
                    )