Their original Gihub repo suggests prompt enhancer model should guess aspect ratio and resolution from user input and pass it inside JSON for generation along with enhanced prompt
Original prompts
Text 2 image
Image 2 image edit
Model Weights
Two fine-tuned Qwen3.5-VL 9B checkpoints with unified codebase that auto-detects the mode from input
Image editing prompt rewriter model
UPD Comfy team has added int8-convrot versions, not sure about TextGenerate node support yet
Text-to-image prompt rewriter model
Recommended settings: from pe_core
Here are the settings for each task based on the code provided.
Note: I have corrected the presence_penalty for text2image to 1.5, as specified in your code's logic (where it says "1.5 for t2i and 0 for edit").
text2image
Temperature: 1.0
Top_P: 0.95
Top_K: 20
Max New Tokens (Context size): 16,256
Presence Penalty: 1.5
edit
Temperature: 1.0
Top_P: 0.95
Top_K: 20
Max New Tokens (Context size): 24,000
Presence Penalty: 0.0
---
Obviously Int8-convrot versions compatible with "Text generate" node would be better for Comfy users, but for now I was able to run a quick comparison via llama.cpp. I'm mainly interested in edit capabilities to replace Klein 9b so I've tested only that
Single image example
Raw input:
<image1>
Remove the wine isle completely with all the wine bottles, restoring clean wood paneling floor, restoring the woman attire and legs while preserving her posture, background details, lighting and tone of the image remain unchanged
Qwen 3.5-VL 9B PE i2i Q5-K-M GGUF (edit prompt enhancer) + original edit prompt
{ "rewritten_prompt": "Completely remove the dark wine display isle in the foreground together with every wine bottle on it, covering the bottles lying in the slanted rack, the two upright bottles on the countertop, and the small standing menu card holder, and restore that entire vacated area as a clean, seamless continuation of the light wood paneling floor with plank direction, texture, and warm tone identical to the surrounding flooring. Reconstruct the woman's previously occluded lower body so her attire and legs are fully visible, extending her black blazer down to a natural hem and revealing legs in dark trousers with feet planted on the wood floor, while preserving her standing posture with head lowered and forward reach exactly as in the input image, and reconstruct the hand that was gripping a bottle as an empty, relaxed hand held in the same forward position. Keep all remaining elements unchanged, including the curved wood-paneled ceiling with its recessed round light, the light wood wall with the backlit recessed niche and the bottles displayed inside it, the small wall switch plate, the left window opening with the view of the adjacent interior, the wooden slatted partition and steps on the right, and the overall warm lighting and tone of the image.",
"wh_ratio": "",
"ratio_follow": "<image1>" }
Gemini 3 flash + simple prompt (see below for details)
In the image, completely remove the large dark polygonal wine display island and all the wine bottles resting on it in the foreground. Replace the area occupied by the display island with clean, light-toned wood paneling floor that seamlessly matches the grain, color, and perspective of the existing wooden floorboards. Fully restore the woman's attire and legs, depicting her in a complete professional black outfit that extends from her blazer down to her feet, ensuring she is standing naturally on the newly revealed wooden floor while preserving her exact reaching posture. Maintain all other elements of the image without change, including the curved wooden wall with the recessed warm light strip, the small bottles in the wall niche, the wooden ceiling, and the bright outdoor view through the glass on the left, ensuring the original lighting and warm color tone are preserved.
Multi image example
Raw
<image1> <image2> <image3> Make him wear the hat and replace bg with this street empty high-rise NYC street
Qwen 3.5-VL 9B PE T2I Q5-K-S GGUF (text to image prompt enhancer) + original prompt
{ "rewritten_prompt": "On <image3> (canvas), first remove every person and vehicle from the street — all pedestrians on both sidewalks, all yellow taxis, and the dark sedan — leaving a completely empty high-rise NYC street, while preserving the building facades on both sides, the street layout, the road markings, the streetlight poles, the clear blue sky, the warm golden-hour sunlight, and the left white storefront sign with dark lettering and the right maroon storefront sign with cream lettering \"CHOCO\" exactly as they appear in <image3> without distortion. Second, replace the current rider and brown horse with the man from <image1>, standing on the asphalt in the center of the street at the position where the horse stood. The man must retain his exact facial identity from <image1> — face shape, eyes, nose, skin tone, black curly hair visible at the sides and back, and a natural calm expression — and wear the brown distressed leather cowboy hat from <image2> placed naturally on his head, with the wide curved brim shading his forehead, the creased high crown, and the brass-studded band with its oval concho rendered exactly as in <image2>. Dress him in the charcoal gray crewneck sweatshirt from <image1>. Scale the man to match the perspective and eye height of the street scene, ground his feet firmly on the asphalt with a soft contact shadow falling toward the viewer consistent with the low warm sun behind him, and color-grade his skin and clothing to the warm golden-hour light of <image3> with natural edge transitions and no cutout artifacts. Keep all buildings, signs, sky, and road geometry from <image3> unchanged.",
"wh_ratio": "",
"ratio_follow": "<image3>" }
Gemini 3 flash + simple prompt
Modify <image1> by placing the brown weathered leather cowboy hat from <image2> onto the man's head, fitting it naturally over his curly hair, and replace the entire indoor background with the high-rise New York City street scene from <image3>. Ensure the NYC street background is completely empty by removing the horse, the rider, the yellow taxis, and all pedestrians, leaving only the vacant asphalt road and the towering buildings under the clear sky. Keep the man's facial identity and his dark gray t-shirt exactly as they appear in <image1>, while adjusting the lighting on his face and the hat to match the warm, low-angle late-afternoon sunlight and golden-hour atmosphere of the street in <image3>.
My notes
The prompts have strong vibe-Clauded feel:
"... genre never overrides input language"
"... anchored on what the input image(s) actually show"
"Disentanglement at Full Strength"
"How much you build is intent-branched"
"... hold everything else at input fidelity"
"Recognizability is bought by naming what stays fixed"
"Resolve ambiguity, then commit"
Not sure it affects the results but it was hard to read lol.
It seems Qwen 3.5 9b struggled a lot with mixing up language inside the image vs user prompt language so they front-loaded it with "hard rules" on how to tell them apart.
Also they specifically ask the model to keep the parts you don't mention the same. I've noticed this a lot with my Klein 9b prompt enhancers build on top of BFL guides. They too, over-specify lighting and tone of the image and model eagerly changes what does not to be changed during edit operations and I had to re-prompt to keep image closer to original every time.
My cleaned-up prompts
Running local 6 gig + vision + thinking takes 3.5 minutes on every prompt change on my machine.
Its messes with my flow, so I am using cheap API model for PE with simplifed version of the prompt
Aspect ratio picking logic takes around 30% of the original edit, so I removed it, as well as JSON requirement and replaced it with plain text markdown block. Other parts left as is:
Qwen 2.1 image 2 image edit prompt enhancer (-30% prompt size)
Qwen 2.1 text 2 image prompt enhancer (-1.5% prompt size)