Extensions/Seedream Image Generate ComfyUI Node
ComfyUI Extension

Seedream Image Generate ComfyUI Node

A ComfyUI custom node based on the Volcano Engine Doubao large model Seedream API, designed for high-quality image generation.

By dzy1128·Created 11 months ago·Updated 26 days ago· 7
dzy1128/Seedream-Image-Generate-ComfyUI
Nodes5
On cloudLocal install
Categoryvideo/generation, image/generation
Stars7
Updated26 days ago
Readme

Seedream Image Generate ComfyUI Node

一个基于火山引擎豆包大模型Seedream API的ComfyUI自定义节点,用于高质量图像生成。

<!-- 使用示例:添加节点界面截图 ![节点界面](images/screenshots/node-interface.png) -->

激活模型

  1. 访问火山模型广场
  2. 在<图片生成>下面找到Seedream-4.0模型,鼠标悬浮在模型上面,会出现以下界面,然后点击API接入:

Seedream-4.0模型激活

  1. 选择你的API_KEY,点击<选择使用>,然后点击开通模型。

功能特性

  • 🎨 多模型支持: 支持doubao-seedream系列模型
  • 🖼️ 多图像输入: 支持最多5张输入图像(1张必选,4张可选)
  • 📐 灵活宽高比: 支持常见宽高比选择(1:1, 2:3, 3:2, 4:3, 3:4, 16:9, 9:16, 21:9)
  • 🔄 顺序图像生成: 支持自动、启用、禁用模式
  • ⚙️ 完全可配置: 所有API参数都可以在节点中配置
  • 🎯 多图输出: 支持一次生成多张图像
  • 📄 详细信息输出: 提供生成过程的详细文本信息

安装要求

  1. 确保已安装ComfyUI
  2. 安装依赖包:
pip install volcengine-python-sdk[ark]

配置

获取API密钥

  1. 访问火山引擎控制台
  2. 开通ARK服务并获取API Key
  3. 设置环境变量:
export ARK_API_KEY="your_api_key_here"

安装节点

  1. 将此文件夹复制到ComfyUI的custom_nodes目录
  2. 重启ComfyUI
  3. 在节点菜单中找到 "image/generation" → "Seedream Image Generate"

节点参数说明

必需参数

  • prompt: 图像生成提示词(支持中英文)
  • model: 选择生成模型
    • doubao-seedream-4-0-250828 (默认)
    • doubao-seedream-4-5-251128
  • aspect_ratio: 图像宽高比
    • 1:1 (2048x2048), 2:3 (1664x2496), 3:2 (2496x1664), 4:3 (2304x1728), 3:4 (1728x2304), 16:9 (2560x1440), 9:16 (1440x2560), 21:9 (3024x1296), 2K, 3K, 3.5K, 4K

输出参数

  • images: 生成的图像列表
  • text: 详细的生成信息文本,包括:
    • 生成参数信息
    • 每张图像的URL和尺寸
    • 模型返回的元数据
    • 错误信息(如果生成失败)

可选参数

图像输入

  • image1-image5: 可选的输入图像(支持0-5张)
    • 不提供图像时为文生图模式
    • 提供图像时为图生图模式

顺序生成控制(多图生成)

  • sequential_image_generation: 顺序生成模式
    • auto - 自动(默认)
    • enabled - 启用
    • disabled - 禁用
  • max_images: 最大生成图像数量 (1-10)
    • 对应官方API的 sequential_image_generation_options.max_images
    • 用于控制顺序生成时的图片数量
    • 示例:设置为4时,API会返回最多4张图片
    • ⚠️ 重要:要生成多张图片,必须同时启用 stream=True
  • stream: 流式传输
    • True - 启用流式传输(生成多张图片时必须启用
    • False - 禁用(默认,只返回1张图片)

其他参数

  • response_format: 响应格式 (url/b64_json)
  • watermark: 是否添加水印
  • stream: 是否使用流式传输
  • base_url: API基础URL
  • use_local_images: 启用本地图像Base64编码(默认开启,官方支持)
  • seed: 种子值(用于工作流跟踪,支持大整数)
  • enable_auto_retry: 启用自动重试机制(默认开启,处理云端工作流异步问题)

使用示例

<!-- 工作流程示例图片 ![工作流示例](images/workflow/basic-workflow.png) -->
  1. 文生图模式(纯提示词生成)

    • 不连接任何图像输入
    • 输入提示词:"一个美丽的风景画,高清,4K"
    • 选择合适的宽高比(如 16:9)
    • 点击执行
    • 查看images输出的生成图像
    • text输出会显示 "文生图模式"
  2. 图生图模式(基于图像生成)

    • 连接1-5张图像到image1-image5
    • 输入提示词:"转换为油画风格"
    • 选择合适的宽高比
    • 点击执行
    • text输出会显示 "图生图模式"
  3. 顺序生成(一次生成多张图片)

    ⚠️ 关键配置要求

    • 必须启用 stream = True
    • 设置 max_images > 1
    • sequential_image_generation 设为 "enabled""auto"

    配置示例

    sequential_image_generation: "auto"
    max_images: 3
    stream: True  ← 必须!
    

    这会生成对应的API调用:

    {
      "sequential_image_generation": "auto",
      "sequential_image_generation_options": {
        "max_images": 3
      },
      "stream": true
    }
    

    说明

    • max_images 参数会通过SDK自动转换为 API 的 sequential_image_generation_options
    • 内部使用 SequentialImageGenerationOptions(max_images=N) 类进行序列化
    • 节点会通过流式响应收集所有生成的图片(会过滤掉无效的中间数据)
    • 控制台会显示收集进度和有效图片数量
    • 适合需要多个变体的场景

    重要提示

    • 只有有效的图片(包含URL或b64_json)才会被收集
    • 流式响应中的中间状态数据会被自动过滤
    • 最终返回的图片数量可能小于 max_images 设置的值(取决于API实际生成的有效图片数)
<!-- 节点参数截图 ![节点参数](images/screenshots/node-parameters.png) -->
  1. 本地图像使用
    • 启用 use_local_images=True(默认开启)
    • 节点会自动将本地图像转换为Base64格式
    • 支持PNG、JPEG等常见格式,自动转换为PNG
    • 查看text输出了解转换状态和详细信息
<!-- 生成结果示例 <div align="center"> <img src="images/examples/input-image.png" alt="输入图像" width="300"> <img src="images/examples/output-image.png" alt="生成结果" width="300"> <p><em>左:输入图像 | 右:生成结果</em></p> </div> -->

注意事项

  • 确保网络连接稳定,API调用需要网络访问
  • 图像生成可能需要一些时间,请耐心等待
  • API有使用限制,请合理使用避免超出配额
  • 本地图像支持: 现已支持本地图像输入(Base64编码格式)
  • 启用 use_local_images=True(默认)时会自动转换本地图像为Base64格式
  • 如果Base64转换失败,会自动回退到示例图像确保稳定性

故障排除

  1. API Key错误: 确保正确设置ARK_API_KEY环境变量
  2. 网络错误: 确保网络连接正常,可以访问火山引擎服务
  3. 图像加载失败: 检查输入图像格式是否支持
  4. 依赖包问题: 确保已安装 pip install 'volcengine-python-sdk[ark]'

支持与反馈

如果遇到问题或有改进建议,欢迎提交Issue或Pull Request。