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