test p2p

app_haircolor_img2img.py

@@ -77,17 +77,15 @@ def image_to_image(
 
     cond_image = input_image
 
-    # generator = torch.Generator(device=DEVICE).manual_seed(seed)
+    generator = torch.Generator(device=DEVICE).manual_seed(seed)
     generated_image = basepipeline(
-        seeds=[seed],
-        # generator=generator,
+        generator=generator,
         prompt=edit_prompt,
         negative_prompt=DEFAULT_NEGATIVE_PROMPT,
         image=cond_image,
         height=generate_size,
         width=generate_size,
         guidance_scale=guidance_scale,
-        image_guidance_scale=1.5,
         num_inference_steps=num_steps,
     ).images[0]