From f56c10c26f0a0bb95225724f34417b1ef309779e Mon Sep 17 00:00:00 2001 From: liubo0902 <38622806+liubo0902@users.noreply.github.com> Date: Wed, 13 Sep 2023 08:29:17 +0800 Subject: [PATCH] Update batch_hijack.py --- scripts/batch_hijack.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/batch_hijack.py b/scripts/batch_hijack.py index 79804e286..dbd971232 100644 --- 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])