test control img2img

app_haircolor_img2img.py

@@ -74,7 +74,7 @@ def image_to_image(
     seed: int,
     num_steps: int,
     guidance_scale: float,
-    image_guidance_scale: float,
+    strength: float,
     generate_size: int,
     cond_scale1: float = 1.2,
     cond_scale2: float = 1.2,
@@ -97,7 +97,7 @@ def image_to_image(
         height=generate_size,
         width=generate_size,
         guidance_scale=guidance_scale,
-        image_guidance_scale=image_guidance_scale,
+        strength=strength,
         num_inference_steps=num_steps,
         controlnet_conditioning_scale=[cond_scale1, cond_scale2],
     ).images[0]
@@ -137,7 +137,7 @@ def create_demo() -> gr.Blocks:
             with gr.Column():
                 num_steps = gr.Slider(minimum=1, maximum=100, value=20, step=1, label="Num Steps")
                 guidance_scale = gr.Slider(minimum=0, maximum=30, value=5, step=0.5, label="Guidance Scale")
-                image_guidance_scale = gr.Slider(minimum=0, maximum=30, value=1.5, step=0.1, label="Image Guidance Scale")
+                strength = gr.Slider(minimum=0, maximum=3, value=0.2, step=0.1, label="Strength")
             with gr.Column():
                 with gr.Accordion("Advanced Options", open=False):
                     mask_expansion = gr.Number(label="Mask Expansion", value=50, visible=True)
@@ -164,7 +164,7 @@ def create_demo() -> gr.Blocks:
             outputs=[origin_area_image, croper],
         ).success(
             fn=image_to_image,
-            inputs=[origin_area_image, edit_prompt,seed, num_steps, guidance_scale, image_guidance_scale, generate_size, cond_scale1, cond_scale2],
+            inputs=[origin_area_image, edit_prompt,seed, num_steps, guidance_scale, strength, generate_size, cond_scale1, cond_scale2],
             outputs=[generated_image, generated_cost],
         ).success(
             fn=restore_result,