xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-06 14:07:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: image.shape accessed before image is null-checked (#134)

Browse Source
This commit is contained in:
Christian Byrne 2025-05-05 02:18:24 -07:00 committed by GitHub
parent c0a348d778
commit 0b0eead4ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_api_nodes/nodes_openai.py
View File

@ -427,10 +427,10 @@ class OpenAIGPTImage1(ComfyNodeABC):
files.append(("image[]", img_binary)) files.append(("image[]", img_binary))
if mask is not None: if mask is not None:
if image.shape[0] != 1:
raise Exception("Cannot use a mask with multiple image")
if image is None: if image is None:
raise Exception("Cannot use a mask without an input image") raise Exception("Cannot use a mask without an input image")
if image.shape[0] != 1:
raise Exception("Cannot use a mask with multiple image")
if mask.shape[1:] != image.shape[1:-1]: if mask.shape[1:] != image.shape[1:-1]:
raise Exception("Mask and Image must be the same size") raise Exception("Mask and Image must be the same size")
batch, height, width = mask.shape batch, height, width = mask.shape