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

fix ruff

Browse Source
This commit is contained in:
thot-experiment 2025-04-23 13:01:09 -07:00
parent a25c2f2e2a
commit dc049d03cd
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
comfy_api_nodes/nodes_api.py
View File

@ -342,26 +342,26 @@ class OpenAIGPTImage1(ComfyNodeABC):
img_binaries = [] img_binaries = []
mask_binary = None mask_binary = None
files = [] 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
batch_size = image.shape[0] batch_size = image.shape[0]
for i in range(batch_size): for i in range(batch_size):
single_image = image[i:i+1] single_image = image[i:i+1]
scaled_image = downscale_input(single_image).squeeze() 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 img_binary = img_byte_arr
img_binary.name = f"image_{i}.png" img_binary.name = f"image_{i}.png"
img_binaries.append(img_binary) img_binaries.append(img_binary)
if batch_size == 1: if batch_size == 1:
files.append(("image", img_binary)) files.append(("image", img_binary))