wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-29 07:40:21 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

Populate from env var whenever it is set

Browse Source
This commit is contained in:
doctorpangloss 2025-01-22 13:58:34 -08:00
parent e5044799da
commit 2826d9b4e5
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes/nodes_ideogram.py
View File

@ -1,4 +1,5 @@
import json import json
import os
from io import BytesIO from io import BytesIO
from itertools import chain from itertools import chain
from typing import Tuple, Dict, Any from typing import Tuple, Dict, Any
@ -12,7 +13,6 @@ from comfy.nodes.package_typing import CustomNode
from comfy.utils import pil2tensor, tensor2pil from comfy.utils import pil2tensor, tensor2pil
from comfy_extras.constants.resolutions import IDEOGRAM_RESOLUTIONS from comfy_extras.constants.resolutions import IDEOGRAM_RESOLUTIONS
from comfy_extras.nodes.nodes_mask import MaskToImage from comfy_extras.nodes.nodes_mask import MaskToImage
from comfy.cli_args import args
ASPECT_RATIOS = [(10, 6), (16, 10), (9, 16), (3, 2), (4, 3)] ASPECT_RATIOS = [(10, 6), (16, 10), (9, 16), (3, 2), (4, 3)]
ASPECT_RATIO_ENUM = ["ASPECT_1_1"] + list(chain.from_iterable( ASPECT_RATIO_ENUM = ["ASPECT_1_1"] + list(chain.from_iterable(
@ -24,12 +24,12 @@ AUTO_PROMPT_ENUM = ["AUTO", "ON", "OFF"]
RESOLUTION_ENUM = [f"RESOLUTION_{w}_{h}" for w, h in IDEOGRAM_RESOLUTIONS] RESOLUTION_ENUM = [f"RESOLUTION_{w}_{h}" for w, h in IDEOGRAM_RESOLUTIONS]
def api_key_in_env_or_workflow(api_key_from_workflow: str): def api_key_in_env_or_workflow(api_key_from_workflow: str):
if api_key_from_workflow is not None and "" != api_key_from_workflow: from comfy.cli_args import args
if api_key_from_workflow is not None and "" != api_key_from_workflow.strip():
return api_key_from_workflow return api_key_from_workflow
return args.ideogram_api_key return os.environ.get("IDEOGRAM_API_KEY", args.ideogram_api_key)
class IdeogramGenerate(CustomNode): class IdeogramGenerate(CustomNode):