ComfyUI-Prompt-Expansion

ComfyUI-Prompt-Expansion
★ 80

提示扩展本地GPT-2ComfyUI节点提示工程
在本地用GPT-2动态扩展提示词，将输入的正向prompt生成并合并为更丰富的输出，方便快速构建多样化提示。
💡 将简单正向prompt扩展为丰富的生成提示。
🍴 16 Forks💻 Python🔄 2024-05-22
📦
网盘下载
复制链接后前往夸克网盘下载
https://pan.quark.cn/s/c73fe210bae7
📄 README

ComfyUI Prompt Expansion

Dynamic prompt expansion, powered by GPT-2 locally on your device.

The custom node will analyze your Positive prompt and Seed and incorporate additional keywords, which will likely improve your resulting image.

For example, (from the workflow image below):

Original prompt: “Portrait of robot Terminator, cybord, evil, in dynamics, highly detailed, packed with hidden details, style, high dynamic range, hyper realistic, realistic attention to detail, highly detailed”

Expanded prompt: “Portrait of robot Terminator, cybord, evil, in dynamics, highly detailed, packed with hidden details, style, high dynamic range, hyper realistic, realistic attention to detail, highly detailed, extremely lifelike, digital painting, artstation, illustration, concept art, smooth, sharp focus, 8k“

Based on my testing, it’s evident that incorporating additional keywords generated by GPT-2 notably improves the quality of the resulting images.

Many thanks to the authors of the Fooocus project for generously providing most of the code.

Installation

  • To install this custom node for ComfyUI, clone the repository using Git or download it, and then extract the the files to: ComfyUI\custom_nodes\ComfyUI-Prompt-Expansion:
    • https://github.com/meap158/ComfyUI-Prompt-Expansion.git

    Usage

    To use this custom node (located within the ‘utils’ submenu), simply connect your positive prompt to it, which will then output the joined prompt.

    You can also load the example workflow by dragging the workflow file (workflow_example.json or workflow_example.png) onto ComfyUI.

    Inputs

  • text: Your positive prompt.
  • log_prompt: Whether to log the Prompt Expansion suffix and the final prompt to the console.

    • Note

    Upon the first launch, the custom node will automatically download the model file (343MB) from the following link:

    https://huggingface.co/lllyasviel/misc/resolve/main/fooocus_expansion.bin