Prompt Expansion - Custom Node for ComfyUI

由本地设备上的 GPT-2 驱动的动态提示词扩展功能。

此自定义节点将分析您的“正向提示词”和“种子值”，并添加额外的关键词，这很可能提升生成图像的质量。

例如（来自上方图像）：

原始提示词：机器人终结者肖像，赛博格，邪恶，动态感，高度细节，充满隐藏细节，风格，高动态范围，超写实，真实细节刻画，高度详细

扩展后提示词：机器人终结者肖像，赛博格，邪恶，动态感，高度细节，充满隐藏细节，风格，高动态范围，超写实，真实细节刻画，高度详细， 极其逼真，数字绘画，ArtStation，插画，概念艺术，平滑，锐利聚焦，8K

根据我的测试，很明显，GPT-2 生成的附加关键词显著提升了生成图像的质量。

非常感谢 Fooocus 项目作者慷慨提供了大部分代码。

如果您觉得我的工作有帮助，请考虑买杯咖啡支持我！谢谢！https://www.buymeacoffee.com/meap

安装

要安装此 ComfyUI 自定义节点，请下载或使用 Git 克隆仓库，然后将文件解压至：ComfyUI\custom_nodes\ComfyUI-Prompt-Expansion

使用方法

要使用此自定义节点（位于 'utils' 子菜单中），只需将您的正向提示词连接到它，它将输出合并后的提示词。
您也可以通过将工作流文件（workflow_example.json 或 workflow_example.png）拖拽到 ComfyUI 中来加载示例工作流。

输入项

注意

首次运行时，该自定义节点将自动从以下链接下载模型文件（343MB）：