fix controlnet input_mode ENUM

lllyasviel · Feb 29, 2024 · 715f9c8 · 715f9c8
1 parent b59deaa
commit 715f9c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
@@ -155,7 +155,7 @@ def get_input_data(self, p, unit, preprocessor, h, w):
         image_list = []
         resize_mode = external_code.resize_mode_from_value(unit.resize_mode)
 
-        if unit.input_mode == external_code.InputMode.MERGE:
+        if unit.input_mode == external_code.InputMode.MERGE.value:
             for idx, item in enumerate(unit.batch_input_gallery):
                 img_path = item['name']
                 logger.info(f'Try to read image: {img_path}')
@@ -169,7 +169,7 @@ def get_input_data(self, p, unit, preprocessor, h, w):
                     mask = np.ascontiguousarray(cv2.imread(mask_path)[:, :, ::-1]).copy()
                 if img is not None:
                     image_list.append([img, mask])
-        elif unit.input_mode == external_code.InputMode.BATCH:
+        elif unit.input_mode == external_code.InputMode.BATCH.value:
             image_list = []
             image_extensions = ['.jpg', '.jpeg', '.png', '.bmp']
             batch_image_files = shared.listfiles(unit.batch_image_dir)