From 9ee746ef363a2db994fd9103811c54e175d6fb5b Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Mon, 15 Jan 2024 07:14:07 -0500
Subject: [PATCH] faceid fix batch

---
 scripts/faceid.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/scripts/faceid.py b/scripts/faceid.py
index 7aa1846a5..0b739832d 100644
--- a/scripts/faceid.py
+++ b/scripts/faceid.py
@@ -178,7 +178,11 @@ class Script(scripts.Script):
         ip_model_dict['faceid_embeds'] = face_embeds
 
         # run generate
-        images = ip_model.generate(**ip_model_dict)
+        images = []
+        for _i in range(p.n_iter):
+            res = ip_model.generate(**ip_model_dict)
+            if isinstance(res, list):
+                images += res
 
         if not cache:
             ip_model = None