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

Ruff.

Browse Source
This commit is contained in:
Robin Huang 2025-04-22 10:20:50 -07:00
parent daf1fab89b
commit 3ae7ecfe82
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_api_nodes/nodes_api.py
View File

@ -100,7 +100,6 @@ class IdeogramTextToImage(ComfyNodeABC):
def api_call(self, prompt, model, aspect_ratio=None, resolution=None, def api_call(self, prompt, model, aspect_ratio=None, resolution=None,
magic_prompt_option="AUTO", seed=0, style_type="NONE", magic_prompt_option="AUTO", seed=0, style_type="NONE",
negative_prompt="", num_images=1, color_palette="", auth_token=None): negative_prompt="", num_images=1, color_palette="", auth_token=None):
import io
import numpy as np import numpy as np
import requests import requests
@ -237,13 +236,12 @@ class OpenAITextToImage(ComfyNodeABC):
elif model == "dall-e-3": elif model == "dall-e-3":
if size == "auto": if size == "auto":
size = "1024x1024" size = "1024x1024"
valid_sizes = ["1024x1024", "1792x1024", "1024x1792"] valid_sizes = ["1024x1024", "1792x1024", "1024x1792"]
if size not in valid_sizes: if size not in valid_sizes:
raise ValueError(f"Size {size} not valid for dall-e-3. Must be one of: {', '.join(valid_sizes)}") raise ValueError(f"Size {size} not valid for dall-e-3. Must be one of: {', '.join(valid_sizes)}")
# TODO: add NEW MODEL # TODO: add NEW MODEL
import io
import numpy as np import numpy as np
import torch import torch
@ -271,7 +269,7 @@ class OpenAITextToImage(ComfyNodeABC):
response = operation.execute() response = operation.execute()
# validate raw JSON response # validate raw JSON response
data = response.data data = response.data
if not data or len(data) == 0: if not data or len(data) == 0:
raise Exception("No images returned from OpenAI endpoint") raise Exception("No images returned from OpenAI endpoint")