Update diffusion_webui/diffusion_models/img2img_app.py

diffusion_webui/diffusion_models/img2img_app.py

@@ -21,6 +21,7 @@ class StableDiffusionImage2ImageGenerator:
             )
             
             self.pipe.model_name = stable_model_path
+            self.pipe.scheduler_name = scheduler
             self.pipe = get_scheduler(pipe=self.pipe, scheduler=scheduler)
             self.pipe.to("cuda")
             self.pipe.enable_xformers_memory_efficient_attention()