faceid fix batch

This commit is contained in:
Vladimir Mandic
2024-01-15 07:14:07 -05:00
parent 6bdc725407
commit 9ee746ef36
+5 -1
View File
@@ -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