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/instruct_pix2pix.py
+++ b/lama_cleaner/model/instruct_pix2pix.py
@@ -59,7 +59,7 @@ class InstructPix2Pix(DiffusionInpaintModel):
            num_inference_steps=config.p2p_steps,
            image_guidance_scale=config.p2p_image_guidance_scale,
            guidance_scale=config.p2p_guidance_scale,
-            output_type="np.array",
+            output_type="np",
            generator=torch.manual_seed(config.sd_seed)
        ).images[0]