Minor bug fix - checking type of combinatorial batches

adieyal · Nov 12, 2022 · a729de9 · a729de9
1 parent 798901e
commit a729de9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/CHANGELOG b/CHANGELOG
@@ -32,4 +32,5 @@
 0.15.1 Fixed an error caused when setting wildcard_dir in config.json, also other minor bug fixes
 0.15.2 Handle situation where no prompts are generated.
 0.16.0 Added a checkbox to enable writing all generated prompts to a text file.
-0.17.0 Dynamic prompts is now an AlwaysVisible script. It now plays nicely with other scripts like X/Y plot. Big thanks to @stysmmaker
+0.17.0 Dynamic prompts is now an AlwaysVisible script. It now plays nicely with other scripts like X/Y plot. Big thanks to @stysmmaker
+0.17.1 Minor bug fix checking type of combinatorial batches
diff --git a/scripts/dynamic_prompting.py b/scripts/dynamic_prompting.py
@@ -34,7 +34,7 @@
 else:
     WILDCARD_DIR = Path(wildcard_dir)
 
-VERSION = "0.17.0"
+VERSION = "0.17.1"
 
 
 wildcard_manager = WildcardManager(WILDCARD_DIR)
@@ -123,7 +123,12 @@ def process(
 
         original_prompt = p.prompt[0] if type(p.prompt) == list else p.prompt
         original_seed = p.seed
-        if combinatorial_batches < 1:
+
+        try:
+            combinatorial_batches = int(combinatorial_batches)
+            if combinatorial_batches < 1:
+                combinatorial_batches = 1
+        except (ValueError, TypeError):
             combinatorial_batches = 1
 
         if is_combinatorial: