From 14ddfb43bb26cf539613748bf7ec78f3e2508ccb Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Tue, 14 Nov 2023 20:32:23 -0600
Subject: [PATCH] fix(api): handle empty prompts when parsing

---
 api/onnx_web/diffusers/utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
index 651e861a..7eaee674 100644
--- a/api/onnx_web/diffusers/utils.py
+++ b/api/onnx_web/diffusers/utils.py
@@ -515,6 +515,9 @@ def parse_reseed(prompt: str) -> Tuple[str, List[Reseed]]:
     return get_tokens_from_prompt(prompt, RESEED_TOKEN, parser=parse_reseed_group)
 
 
-def remove_tokens(prompt: str) -> str:
+def remove_tokens(prompt: Optional[str]) -> Optional[str]:
+    if prompt is None:
+        return prompt
+
     remainder, tokens = get_tokens_from_prompt(prompt, ANY_TOKEN)
     return remainder