ComfyUI Extension: ComfyUI-Prompt-Expansion

ComfyUI-Prompt-Expansion Introduction

ComfyUI-Prompt-Expansion is a powerful extension designed to enhance your creative process by dynamically expanding your prompts using GPT-2, a state-of-the-art language model. This extension works locally on your device, ensuring privacy and efficiency. By analyzing your initial prompt and seed, it generates additional keywords that can significantly improve the quality and detail of your resulting images. This tool is particularly useful for AI artists looking to add depth and complexity to their artwork without manually brainstorming every detail.

How ComfyUI-Prompt-Expansion Works

At its core, ComfyUI-Prompt-Expansion leverages the capabilities of GPT-2 to enrich your prompts. Think of GPT-2 as a highly intelligent assistant that understands the nuances of language and can suggest relevant additions to your prompts. When you input a basic prompt, the extension processes it and suggests additional keywords that align with the original theme. For example, if your initial prompt is "Portrait of robot Terminator, cyborg, evil, in dynamics, highly detailed," the extension might expand it to include terms like "extremely lifelike, digital painting, artstation, illustration, concept art, smooth, sharp focus, 8k." These additions help create more detailed and visually appealing images.

ComfyUI-Prompt-Expansion Features

Dynamic Prompt Expansion

This feature automatically enhances your prompts by adding relevant keywords. It saves you time and effort by generating detailed and contextually appropriate expansions.

Local Processing

All the processing is done locally on your device, ensuring that your data remains private and secure. There's no need to worry about your prompts being sent to external servers.

Custom Node Integration

The extension integrates seamlessly with ComfyUI as a custom node. You can easily connect your positive prompt to this node, and it will output the expanded prompt.

Example Workflows

To help you get started, the extension includes example workflows. You can load these by dragging the workflow file (workflow_example.json or workflow_example.png) onto ComfyUI.

Logging Options

You have the option to log the expanded prompt and the final prompt to the console. This can be useful for keeping track of the changes and understanding how the extension enhances your prompts.

ComfyUI-Prompt-Expansion Models

The extension uses a specific model file to perform the prompt expansion. Upon the first launch, it will automatically download this model file (343MB) from the following link:

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

This model is pre-trained and optimized for generating high-quality prompt expansions. There are no multiple models to choose from, simplifying the user experience.

Troubleshooting ComfyUI-Prompt-Expansion

Common Issues and Solutions

Issue: The extension is not expanding my prompts.

Solution: Ensure that you have connected your positive prompt to the custom node correctly. Check if the model file has been downloaded successfully. If not, try downloading it manually from the provided link.

Issue: The expanded prompt is not appearing in the console.

Solution: Make sure that the log_prompt option is enabled. This setting allows the expanded prompt to be logged to the console for your review.

Issue: The extension is not working as expected after installation.

Solution: Verify that you have placed the extension files in the correct directory: ComfyUI\custom_nodes\ComfyUI-Prompt-Expansion. Restart ComfyUI to ensure that the extension is loaded properly.

Frequently Asked Questions

Q: Do I need an internet connection to use this extension? A: An internet connection is only required for the initial download of the model file. After that, all processing is done locally.

Q: Can I customize the keywords added by the extension? A: The extension automatically generates keywords based on your initial prompt. While you cannot directly customize the keywords, you can influence the output by carefully crafting your initial prompt.

Learn More about ComfyUI-Prompt-Expansion

To further enhance your experience with ComfyUI-Prompt-Expansion, consider exploring the following resources:

• ComfyUI Documentation: Comprehensive documentation for ComfyUI, including how to use custom nodes.
• GPT-2 Overview: Learn more about the GPT-2 model and its capabilities.
• Community Forums: Join forums and communities where you can ask questions, share your experiences, and get support from other AI artists. By leveraging these resources, you can maximize the potential of ComfyUI-Prompt-Expansion and take your AI art to the next level.