Add denoise parameter

2024-10-20 14:52:31 +01:00
parent 712f0106b9
commit 12d39e85ee
1 changed files with 6 additions and 1 deletions
@@ -185,6 +185,10 @@ class DrawThingsImg2Img:
                    {"default": "Euler A Trailing"},
                ),
                "steps": ("INT", {"default": 20, "min": 1, "max": 150, "step": 1}),
+                "denoise": (
+                    "FLOAT",
+                    {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01},
+                ),
            }
        }

@@ -193,7 +197,7 @@ class DrawThingsImg2Img:
    FUNCTION = "generate_image"

    def generate_image(
-        self, images, model, prompt, seed, guidance_scale, sampler, steps
+        self, images, model, prompt, seed, guidance_scale, sampler, steps, denoise
    ):
        # Call the Draw Things API
        api_url = "http://127.0.0.1:7860/sdapi/v1/img2img"
@@ -215,6 +219,7 @@ class DrawThingsImg2Img:
            "sampler": sampler,
            "steps": steps,
            "init_images": encoded_images,
+            "strength": denoise,
        }

        #        response = requests.post(api_url, json=payload)