diff --git a/lama_cleaner/model/kandinsky.py b/lama_cleaner/model/kandinsky.py
index cf583bd..ea9d9c9 100644
--- a/lama_cleaner/model/kandinsky.py
+++ b/lama_cleaner/model/kandinsky.py
@@ -24,13 +24,6 @@ class Kandinsky(DiffusionInpaintModel):
             "torch_dtype": torch_dtype,
         }
 
-        # self.pipe_prior = KandinskyPriorPipeline.from_pretrained(
-        #     self.prior_name, **model_kwargs
-        # ).to("cpu")
-        #
-        # self.model = KandinskyInpaintPipeline.from_pretrained(
-        #     self.model_name, **model_kwargs
-        # ).to(device)
         self.model = AutoPipelineForInpainting.from_pretrained(
             self.model_name, **model_kwargs
         ).to(device)
@@ -54,6 +47,7 @@ class Kandinsky(DiffusionInpaintModel):
         mask = mask.astype(np.float32) / 255
         img_h, img_w = image.shape[:2]
 
+        # kandinsky 没有 strength
         output = self.model(
             prompt=config.prompt,
             negative_prompt=config.negative_prompt,
@@ -65,6 +59,7 @@ class Kandinsky(DiffusionInpaintModel):
             guidance_scale=config.sd_guidance_scale,
             output_type="np",
             callback=self.callback,
+            generator=generator,
         ).images[0]
 
         output = (output * 255).round().astype("uint8")