Update inpainting_module.py

inpainting_module.py

@@ -612,19 +612,32 @@ class InpaintingModule:
         strength: float,
         generator: torch.Generator
     ) -> Image.Image:
-        """Generate using pure SDXL Inpainting pipeline."""
-        with torch.inference_mode():
-            result = self._pipeline(
-                prompt=prompt,
-                negative_prompt=negative_prompt,
-                image=image,
-                mask_image=mask,
-                num_inference_steps=num_steps,
-                guidance_scale=guidance_scale,
-                strength=strength,
-                generator=generator
-            )
-        return result.images[0]
+        """Generate using pure SDXL Inpainting pipeline with DPM++ scheduler for speed."""
+        # Use DPM++ 2M Karras scheduler for faster generation
+        original_scheduler = self._pipeline.scheduler
+        self._pipeline.scheduler = DPMSolverMultistepScheduler.from_config(
+            self._pipeline.scheduler.config,
+            use_karras_sigmas=True,
+            algorithm_type="dpmsolver++"
+        )
+        logger.info("Switched to DPM++ 2M Karras scheduler for Pure Inpainting")
+
+        try:
+            with torch.inference_mode():
+                result = self._pipeline(
+                    prompt=prompt,
+                    negative_prompt=negative_prompt,
+                    image=image,
+                    mask_image=mask,
+                    num_inference_steps=num_steps,
+                    guidance_scale=guidance_scale,
+                    strength=strength,
+                    generator=generator
+                )
+            return result.images[0]
+        finally:
+            # Restore original scheduler
+            self._pipeline.scheduler = original_scheduler
 
     def _generate_controlnet_inpaint(
         self,
@@ -663,4 +676,4 @@ class InpaintingModule:
             "mode": self._current_mode,
             "conditioning_type": self._current_conditioning_type,
             "model_key": self._current_model_key,
-        }
+        }