test p2p

app_haircolor_img2img.py

@@ -77,13 +77,15 @@ def image_to_image(
     time_cost_str = ''
     run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
 
+    cond_image = [input_image]
+
     # generator = torch.Generator(device=DEVICE).manual_seed(seed)
     generated_image = basepipeline(
         seeds=[seed],
         # generator=generator,
         prompt=edit_prompt,
         negative_prompt=DEFAULT_NEGATIVE_PROMPT,
-        image=[input_image],
+        image=cond_image,
         height=generate_size,
         width=generate_size,
         guidance_scale=guidance_scale,
@@ -124,7 +126,7 @@ def create_demo() -> gr.Blocks:
                 edit_prompt = gr.Textbox(lines=1, label="Edit Prompt", value=DEFAULT_EDIT_PROMPT)
                 generate_size = gr.Number(label="Generate Size", value=512)
             with gr.Column():
-                num_steps = gr.Slider(minimum=1, maximum=100, value=25, step=1, label="Num Steps")
+                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")
             with gr.Column():
                 with gr.Accordion("Advanced Options", open=False):