From 4b03caf8b2ea44af9c8ca094ebb1c98317eb622c Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Thu, 28 Mar 2024 09:31:12 -0400
Subject: [PATCH] handle bad prompts

---
 modules/prompt_parser.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/modules/prompt_parser.py b/modules/prompt_parser.py
index 93a892dcd..785d47714 100644
--- a/modules/prompt_parser.py
+++ b/modules/prompt_parser.py
@@ -121,9 +121,15 @@ def get_learned_conditioning_prompt_schedules(prompts, steps):
         class AtStep(lark.Transformer):
             def scheduled(self, args):
                 before, after, _, when = args
-                yield before or () if step <= when else after
+                try:
+                    yield before or () if step <= when else after
+                except StopIteration:
+                    yield ''
             def alternate(self, args):
-                yield next(args[(step - 1)%len(args)]) # pylint: disable=stop-iteration-return
+                try:
+                    yield next(args[(step - 1) % len(args)]) # pylint: disable=stop-iteration-return
+                except StopIteration:
+                    yield ''
             def start(self, args):
                 def flatten(x):
                     if type(x) == str: