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 images above):

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.

If you’ve found my work helpful, consider buying a coffee to show your support. Thank you! https://www.buymeacoffee.com/meap

Installation

To install this custom node for ComfyUI, download it or clone the repository using Git and then extract 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