Improve speed with better scheduler

image_transformation.py

@@ -14,7 +14,7 @@ if is_vision_available():
     from PIL import Image
 
 if is_diffusers_available():
-    from diffusers import DiffusionPipeline
+    from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
 
 
 IMAGE_TRANSFORMATION_DESCRIPTION = (
@@ -50,6 +50,7 @@ class ImageTransformationTool(Tool):
             self.device = get_default_device()
 
         self.pipeline = DiffusionPipeline.from_pretrained(self.stable_diffusion)
+        self.pipeline.scheduler = DPMSolverMultistepScheduler.from_config(self.pipeline.scheduler.config)
 
         self.pipeline.to(self.device)
         if self.device.type == "cuda":
@@ -68,4 +69,5 @@ class ImageTransformationTool(Tool):
             prompt + added_prompt,
             image,
             negative_prompt=negative_prompt,
+            num_inference_steps=40,
         ).images[0]