From 311eca00184afa0d4c165fb1e5e59c3bc29267ae Mon Sep 17 00:00:00 2001
From: NetroScript <noreply@enostrion.com>
Date: Tue, 16 Jan 2024 02:08:26 +0100
Subject: [PATCH] FaceID: Fix same images for batch

Update seed, positive prompt and negative prompt for every generation
---
 scripts/faceid.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/scripts/faceid.py b/scripts/faceid.py
index e4dcb7838..35a201eec 100644
--- a/scripts/faceid.py
+++ b/scripts/faceid.py
@@ -102,15 +102,12 @@ def face_id(p: processing.StableDiffusionProcessing, faces, image, model, overri
 
     # main generate dict
     ip_model_dict = {
-        'prompt': p.all_prompts[0],
-        'negative_prompt': p.all_negative_prompts[0],
         'num_samples': p.batch_size,
         'width': p.width,
         'height': p.height,
         'num_inference_steps': p.steps,
         'scale': scale,
         'guidance_scale': p.cfg_scale,
-        'seed': int(p.all_seeds[0]),
         'faceid_embeds': face_embeds.shape,
     }
 
@@ -128,7 +125,15 @@ def face_id(p: processing.StableDiffusionProcessing, faces, image, model, overri
     # run generate
     images = []
     ip_model.set_scale(scale)
-    for _i in range(p.n_iter):
+    for i in range(p.n_iter):
+        # Update prompts and seed for each iteration
+        ip_model_dict.update(
+            {
+                'prompt': p.all_prompts[i],
+                'negative_prompt': p.all_negative_prompts[i],
+                'seed': int(p.all_seeds[i]),
+            }
+        )
         res = ip_model.generate(**ip_model_dict)
         if isinstance(res, list):
             images += res