lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update controlnet.py

Browse Source
This commit is contained in:
lllyasviel
2024-02-04 22:33:02 -08:00
parent a4d64adb32
commit e02b69aa0c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
View File

@@ -148,7 +148,7 @@ class ControlNetForForgeOfficial(scripts.Script):
for item in unit.batch_input_gallery:
img_path = item['name']
logger.info(f'Try to read image: {img_path}')
img = cv2.imread(img_path)
img = np.ascontiguousarray(cv2.imread(img_path)[:, :, ::-1]).copy()
mask = None
if img is not None:
image_list.append([img, mask])
@@ -161,7 +161,7 @@ class ControlNetForForgeOfficial(scripts.Script):
if any(filename.lower().endswith(ext) for ext in image_extensions):
img_path = os.path.join(unit.batch_image_dir, filename)
logger.info(f'Try to read image: {img_path}')
img = cv2.imread(img_path)
img = np.ascontiguousarray(cv2.imread(img_path)[:, :, ::-1]).copy()
mask = None
if img is not None:
image_list.append([img, mask])