diff --git a/lama_cleaner/model/kandinsky.py b/lama_cleaner/model/kandinsky.py
index 769e805..7a89d9a 100644
--- a/lama_cleaner/model/kandinsky.py
+++ b/lama_cleaner/model/kandinsky.py
@@ -60,6 +60,7 @@ class Kandinsky(DiffusionInpaintModel):
             output_type="np",
             callback=self.callback,
             generator=generator,
+            callback_steps=1,
         ).images[0]
 
         output = (output * 255).round().astype("uint8")
diff --git a/lama_cleaner/model/pipeline/pipeline_stable_diffusion_controlnet_inpaint.py b/lama_cleaner/model/pipeline/pipeline_stable_diffusion_controlnet_inpaint.py
index b1d3b87..3dfd840 100644
--- a/lama_cleaner/model/pipeline/pipeline_stable_diffusion_controlnet_inpaint.py
+++ b/lama_cleaner/model/pipeline/pipeline_stable_diffusion_controlnet_inpaint.py
@@ -17,10 +17,8 @@ from typing import Union, List, Optional, Callable, Dict, Any
 
 import torch
 import PIL.Image
-import numpy as np
-from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
 
-from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_controlnet import *
+from diffusers.pipelines.controlnet.pipeline_controlnet import *
 from diffusers.utils import replace_example_docstring
 
 EXAMPLE_DOC_STRING = """