Handle empty generation output

2023-01-18 01:13:50 -03:00 · 2023-01-18 01:13:50 -03:00 · 1c6b43ff8b
parent 1a19c95668
commit 1c6b43ff8b
1 changed files with 1 additions and 1 deletions
--- a/scripts/main.py
+++ b/scripts/main.py
@ -299,7 +299,7 @@ class Main(scripts.Script):

                x_samples_ddim, models = self.process_batch_horde(p, model, nsfw, shared_laion, seed_variation, post_processing, prompts[0], negative_prompts[0], seeds[0])

-                if x_samples_ddim is None:
+                if x_samples_ddim is None or len(x_samples_ddim) == 0:
                    break

                x_samples_ddim = [s.cpu() for s in x_samples_ddim]