Update batch_hijack.py

sdwebui-extensions · Sep 13, 2023 · f56c10c · f56c10c
1 parent 24c90be
commit f56c10c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/batch_hijack.py b/scripts/batch_hijack.py
@@ -213,7 +213,7 @@ def get_cn_batches(p: processing.StableDiffusionProcessing) -> Tuple[bool, List[
     batches = [[] for _ in range(cn_batch_size)]
     for i in range(cn_batch_size):
         for unit in units:
-            if getattr(unit, 'input_mode', InputMode.SIMPLE) == InputMode.SIMPLE:
+            if getattr(unit, 'input_mode', InputMode.SIMPLE) == InputMode.SIMPLE or getattr(unit, 'input_mode', InputMode.SIMPLE) == 'simple':
                 batches[i].append(unit.image)
             else:
                 batches[i].append(unit.batch_images[i])