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

implement multi image prompting for GPTI Image 1

Browse Source
This commit is contained in:
thot-experiment 2025-04-23 12:55:39 -07:00
parent dea1c7474a
commit 54661031d9
1 changed files with 28 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
comfy_api_nodes/nodes_api.py
View File

@ -339,25 +339,38 @@ class OpenAIGPTImage1(ComfyNodeABC):
model = "gpt-image-1" model = "gpt-image-1"
path = "/proxy/openai/images/generations" path = "/proxy/openai/images/generations"
request_class = OpenAIImageGenerationRequest request_class = OpenAIImageGenerationRequest
img_binary = None img_binaries = []
mask_binary = None mask_binary = None
files = []
if image is not None: if image is not None:
path = "/proxy/openai/images/edits" path = "/proxy/openai/images/edits"
request_class = OpenAIImageEditRequest request_class = OpenAIImageEditRequest
scaled_image = downscale_input(image).squeeze() batch_size = image.shape[0]
for i in range(batch_size):
single_image = image[i:i+1]
scaled_image = downscale_input(single_image).squeeze()
image_np = (scaled_image.numpy() * 255).astype(np.uint8) image_np = (scaled_image.numpy() * 255).astype(np.uint8)
img = Image.fromarray(image_np) img = Image.fromarray(image_np)
img_byte_arr = io.BytesIO() img_byte_arr = io.BytesIO()
img.save(img_byte_arr, format='PNG') img.save(img_byte_arr, format='PNG')
img_byte_arr.seek(0) img_byte_arr.seek(0)
img_binary = img_byte_arr#.getvalue() img_binary = img_byte_arr
img_binary.name = "image.png" img_binary.name = f"image_{i}.png"
img_binaries.append(img_binary)
if batch_size == 1:
files.append(("image", img_binary))
else:
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 mask.shape[1:] != image.shape[1:-1]: if mask.shape[1:] != image.shape[1:-1]:
@ -373,14 +386,10 @@ class OpenAIGPTImage1(ComfyNodeABC):
mask_img_byte_arr = io.BytesIO() mask_img_byte_arr = io.BytesIO()
mask_img.save(mask_img_byte_arr, format='PNG') mask_img.save(mask_img_byte_arr, format='PNG')
mask_img_byte_arr.seek(0) mask_img_byte_arr.seek(0)
mask_binary = mask_img_byte_arr#.getvalue() mask_binary = mask_img_byte_arr
mask_binary.name = "mask.png" mask_binary.name = "mask.png"
files.append(("mask", mask_binary))
files = {}
if img_binary:
files["image"] = img_binary
if mask_binary:
files["mask"] = mask_binary
# Build the operation # Build the operation
operation = SynchronousOperation( operation = SynchronousOperation(