kandinsky

2023-08-30 21:30:11 +08:00
parent 7ba8fdbe76
commit 94211a4985
11 changed files with 170 additions and 12 deletions
--- a/lama_cleaner/model/sd.py
+++ b/lama_cleaner/model/sd.py
@@ -147,7 +147,7 @@ class SD(DiffusionInpaintModel):
            mask_image=PIL.Image.fromarray(mask[:, :, -1], mode="L"),
            num_inference_steps=config.sd_steps,
            guidance_scale=config.sd_guidance_scale,
-            output_type="np.array",
+            output_type="np",
            callback=self.callback,
            height=img_h,
            width=img_w,