ComfyUI 转 Python 扩展

在 ComfyUI 中构建工作流，然后直接获得可运行的 Python 脚本。

ComfyUI-to-Python-Extension 可以将可视化工作流转换为可执行脚本，使您能够从节点图过渡到自动化、实验以及可重复的生成任务，而无需手动重新构建一切。

本项目支持：

• 使用 另存为脚本 从 ComfyUI 界面导出
• 使用 CLI 将保存的 API 格式工作流进行转换

安装

根据您希望使用该项目的方式选择相应的安装方法。本项目支持 Python 3.12 及更高版本。

Web UI 扩展（文件 -> 另存为脚本）

为了让 ComfyUI 将本项目识别为扩展，该仓库必须可以通过 ComfyUI 的 custom_nodes 搜索路径被发现。

您可以采用以下任一方式：

1. 直接克隆到 ComfyUI/custom_nodes 目录下：

cd /path/to/ComfyUI/custom_nodes
git clone https://github.com/pydn/ComfyUI-to-Python-Extension.git
cd ComfyUI-to-Python-Extension
uv sync

2. 将仓库保留在其他位置，然后：

• 创建符号链接将其指向 ComfyUI/custom_nodes
• 或者通过 extra_model_paths.yaml 文件将仓库的父目录添加到 ComfyUI 的 custom_nodes 搜索路径中

符号链接示例：

git clone https://github.com/pydn/ComfyUI-to-Python-Extension.git
cd /path/to/ComfyUI/custom_nodes
ln -s /path/to/ComfyUI-to-Python-Extension ComfyUI-to-Python-Extension
cd /path/to/ComfyUI-to-Python-Extension
uv sync

安装完成后，请重启 ComfyUI。

CLI 导出工具 / 生成的脚本

对于 CLI 使用和生成脚本的执行，您可以将仓库放置在任何位置。

git clone https://github.com/pydn/ComfyUI-to-Python-Extension.git
cd ComfyUI-to-Python-Extension
uv sync
export COMFYUI_PATH=/path/to/ComfyUI

COMFYUI_PATH 帮助导出工具和生成的脚本找到 ComfyUI 的代码库。它本身并不会将此仓库注册为 Web UI 的 ComfyUI 扩展。

程序会首先检查 COMFYUI_PATH 是否已设置。如果未设置，则导出工具会回退到在其父目录中搜索名为 ComfyUI 的文件夹。

Web UI 导出

在当前的 ComfyUI 版本中，另存为脚本 通常位于以下菜单路径下：

文件 -> 另存为脚本

该命令会下载一个生成的 .py 文件。当前的 UI 导出使用默认文件名 workflow_api.py，因此可以在 ComfyUI Desktop 中直接运行，而无需依赖 prompt() 函数。

注意事项：

• 不同前端版本的菜单位置可能有所不同
• Web UI 导出使用固定的默认文件名，而不是交互式地询问文件名

CLI 导出

1. 在 ComfyUI 中，如有需要，请启用开发模式选项。
2. 将工作流保存为 API 格式：文件 -> 导出 (API)。
3. 运行导出工具：

uv run python -m comfyui_to_python

选项：

uv run python -m comfyui_to_python \
  --input_file workflow_api.json \
  --output_file workflow_api.py \
  --queue_size 10

如果您更喜欢旧版封装方式，仍然可以使用：

uv run python comfyui_to_python.py

标志说明：

• --input_file：输入的工作流 JSON 文件，默认为 workflow_api.json
• --output_file：输出的 Python 文件，默认为 workflow_api.py
• --queue_size：生成脚本中的默认执行次数，默认为 10

生成的脚本

生成的脚本依赖于一个正常运行的 ComfyUI 运行时环境。

如果仓库不在 ComfyUI 内部，请设置：

export COMFYUI_PATH=/path/to/ComfyUI

生成的脚本是工作流的导出文件，它不会自动将工作流的输入参数转换为命令行参数。

直接从 ComfyUI 界面的 文件 -> 另存为脚本 导出的脚本已经包含了用于拖放式重新导入所需的前端工作流元数据。这些脚本保存的图像可以直接拖放到 ComfyUI 中，并恢复原始的工作流元数据。

生成的脚本在启动时会复用 ComfyUI 的运行时参数解析器，因此常见的 ComfyUI 内存相关标志，如 --highvram、--normalvram、--lowvram、--novram、--cpu 和 --disable-smart-memory，都可以直接传递给导出的 .py 文件。

生命周期注意事项：

• 导出的脚本是一次性的工作流运行程序，而非长期运行的 ComfyUI 提示服务器
• 它们不会实现跨多次服务调用的 Web UI 提示/结果缓存功能
• 导出的 main() 函数现在会在 finally 块中尽力清理 ComfyUI 的模型和缓存
• 如果嵌入式或重复调用的宿主环境希望在每次运行后主动卸载模型，而不是保留以供重用，请设置 COMFYUI_TOPYTHON_UNLOAD_MODELS=1，或者调用 main(unload_models=True)。

故障排除

• 不支持的 Python 版本： 请使用 Python 3.12 或更高版本，然后重新运行 uv sync
• 另存为脚本 未显示： 请检查您当前使用的 ComfyUI 菜单/前端版本，并在 文件 菜单中查找
• 重启后仍看不到 另存为脚本： 确保此仓库可通过 custom_nodes 路径被 ComfyUI 发现，方法是将其克隆到 ComfyUI/custom_nodes 目录下、创建符号链接，或在 extra_model_paths.yaml 文件中添加外部的 custom_nodes 路径
• 保存时使用默认文件名： 如果您希望使用不同的本地文件名，请在下载后重命名 workflow_api.py
• 找不到 ComfyUI： 请设置 COMFYUI_PATH
• 运行时缺少模型或路径： 请确认目标 ComfyUI 安装及其 extra_model_paths.yaml 配置文件是否正确。

ComfyUI-to-Python-Extension 快速上手指南

将 ComfyUI 的可视化工作流一键转换为可执行的 Python 脚本，便于自动化运行、实验复现及批量生成。

环境准备

• 操作系统：Linux / macOS / Windows
• Python 版本：Python 3.12 或更高版本（必须）
• 前置依赖：
  • 已安装并配置好的 ComfyUI 环境。
  • 推荐安装 uv 包管理工具以加速依赖安装（若未安装，可使用 pip install uv）。

安装步骤

根据使用场景选择以下任一方式安装。

方案一：作为 Web UI 插件安装（推荐）

适用于直接在 ComfyUI 界面中通过菜单导出脚本。需将本扩展放入 ComfyUI 的 custom_nodes 目录。

# 进入 ComfyUI 的 custom_nodes 目录
cd /path/to/ComfyUI/custom_nodes

# 克隆仓库
git clone https://github.com/pydn/ComfyUI-to-Python-Extension.git

# 进入目录并同步依赖
cd ComfyUI-to-Python-Extension
uv sync

# 重启 ComfyUI 使插件生效

提示：若不想移动仓库位置，也可创建软链接或将父目录添加至 extra_model_paths.yaml 配置中。

方案二：仅用于命令行导出 (CLI)

适用于独立运行导出工具或执行生成的脚本，无需集成到 Web 界面。

# 克隆仓库到任意位置
git clone https://github.com/pydn/ComfyUI-to-Python-Extension.git
cd ComfyUI-to-Python-Extension

# 同步依赖
uv sync

# 设置 ComfyUI 根目录环境变量（可选，若未设置会自动向上查找）
export COMFYUI_PATH=/path/to/ComfyUI

基本使用

方法 A：通过 Web UI 直接导出（最简单）

1. 在 ComfyUI 中构建好你的工作流。
2. 点击顶部菜单栏的 File -> Save As Script。
3. 浏览器将自动下载名为 workflow_api.py 的可执行脚本。
4. 在终端运行该脚本即可复现工作流：

   python workflow_api.py
   

   支持传递 ComfyUI 原生参数，如 --lowvram 等。

方法 B：通过命令行转换 API 文件

如果你已有保存为 API 格式的工作流 JSON 文件：

1. 在 ComfyUI 中开启开发模式选项（如需），并通过 File -> Export (API) 保存为 workflow_api.json。
2. 运行转换命令：

   uv run python -m comfyui_to_python \
     --input_file workflow_api.json \
     --output_file my_workflow.py \
     --queue_size 10
   

   • --input_file: 输入的 JSON 文件路径。
   • --output_file: 输出的 Python 脚本名称。
   • --queue_size: 生成脚本中的默认执行次数。

运行生成的脚本

生成的脚本依赖完整的 ComfyUI 运行时环境。若脚本无法找到 ComfyUI，请确保设置了环境变量：

export COMFYUI_PATH=/path/to/ComfyUI
python my_workflow.py

注意：生成的脚本为单次运行器，非长期服务进程。若需在嵌入式环境中反复调用并希望每次运行后彻底释放显存，可设置环境变量 COMFYUI_TOPYTHON_UNLOAD_MODELS=1 或在代码中调用 main(unload_models=True)。