update

2023-12-16 13:34:56 +08:00
parent cbe6577890
commit 24e95daac1
16 changed files with 381 additions and 323 deletions
--- a/lama_cleaner/model/instruct_pix2pix.py
+++ b/lama_cleaner/model/instruct_pix2pix.py
@@ -66,9 +66,9 @@ class InstructPix2Pix(DiffusionInpaintModel):
            image=PIL.Image.fromarray(image),
            prompt=config.prompt,
            negative_prompt=config.negative_prompt,
-            num_inference_steps=config.p2p_steps,
+            num_inference_steps=config.sd_steps,
            image_guidance_scale=config.p2p_image_guidance_scale,
-            guidance_scale=config.p2p_guidance_scale,
+            guidance_scale=config.sd_guidance_scale,
            output_type="np",
            generator=torch.manual_seed(config.sd_seed),
        ).images[0]