test p2p

app_haircolor_img2img.py

@@ -83,10 +83,11 @@ def image_to_image(
         # generator=generator,
         prompt=edit_prompt,
         negative_prompt=DEFAULT_NEGATIVE_PROMPT,
-        image=input_image,
+        image=[input_image],
         height=generate_size,
         width=generate_size,
         guidance_scale=guidance_scale,
+        image_guidance_scale=1.5,
         num_inference_steps=num_steps,
     ).images[0]