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

Fix: Dall-e 2 not setting request content-type dynamically (#113)

Browse Source
This commit is contained in:
Christian Byrne 2025-05-02 16:31:04 -07:00 committed by GitHub
parent 7a10383f90
commit b9c03f6d4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_api_nodes/nodes_openai.py
View File

@ -116,11 +116,13 @@ class OpenAIDalle2(ComfyNodeABC):
): ):
model = "dall-e-2" model = "dall-e-2"
path = "/proxy/openai/images/generations" path = "/proxy/openai/images/generations"
content_type = "application/json"
request_class = OpenAIImageGenerationRequest request_class = OpenAIImageGenerationRequest
img_binary = None img_binary = None
if image is not None and mask is not None: if image is not None and mask is not None:
path = "/proxy/openai/images/edits" path = "/proxy/openai/images/edits"
content_type = "multipart/form-data"
request_class = OpenAIImageEditRequest request_class = OpenAIImageEditRequest
input_tensor = image.squeeze().cpu() input_tensor = image.squeeze().cpu()
@ -166,6 +168,7 @@ class OpenAIDalle2(ComfyNodeABC):
if img_binary if img_binary
else None else None
), ),
content_type=content_type,
auth_token=auth_token, auth_token=auth_token,
) )
@ -401,7 +404,7 @@ class OpenAIGPTImage1(ComfyNodeABC):
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
content_type="multipart/form-data", content_type ="multipart/form-data"
batch_size = image.shape[0] batch_size = image.shape[0]