PromptEnhancer/PromptEnhancer-32B

PromptEnhancerV2 (32B)

PromptEnhancerV2 is a multimodal language model fine-tuned for text-to-image prompt enhancement and rewriting. It restructures user input prompts while preserving the original intent, producing clearer, layered, and logically consistent prompts suitable for downstream image generation tasks.

Model Details

Model Description

PromptEnhancerV2 is a specialized text-to-image prompt rewriting model that employs chain-of-thought reasoning to enhance user prompts.

• Model type: Vision-Language Model for Prompt Enhancement
• Language(s) (NLP): Chinese (zh), English (en)
• License: Apache-2.0
• Finetuned from model: Qwen/Qwen2.5-VL-32B-Instruct

How to Get Started with the Model

• 1. Clone the repository::

git clone https://github.com/ximinng/PromptEnhancer.git
cd PromptEnhancer
pip install -r requirements.txt

• 2. Model Download:

huggingface-cli download PromptEnhancer/PromptEnhancer-32B --local-dir ./models/promptenhancer-32b

• 3. Use the model:

from inference.prompt_enhancer_v2 import PromptEnhancerV2

# Initialize the model
models_root_path = "./models/promptenhancer-32b"
enhancer = PromptEnhancerV2(models_root_path=models_root_path, device_map="auto")

# Enhance a prompt (Chinese or English)
user_prompt = "韩系插画风女生头像，粉紫色短发+透明感腮红，侧光渲染。"
enhanced_prompt = enhancer.predict(
    prompt_cot=user_prompt,
    device="cuda"
)

print("Enhanced:", enhanced_prompt)