AI Generation Assist Tools

Elevate your generative AI results. WaveSpeed provides a complete suite of specialized assist tools designed to refine inputs and polish outputs. From intelligent prompt engineering to professional-grade upscaling and face restoration, optimize every stage of your creative workflow to achieve production-ready quality.

Explore Assist Tools View API Docs

Specialized Assist Models

Improve your generation quality with these essential auxiliary tools.

1. Prompt Engineering (Input Optimization)

Automated Prompt Expansion

Powered by MagicPrompt / Promptist. Turn simple keywords like "cat in space" into detailed, descriptive prompts optimized for diffusion models. It adds lighting, style, and texture descriptors automatically to ensure high-quality output. Best for users struggling with "blank page syndrome" or getting generic results. Maximize output quality from models like FLUX and those in our Best Open Source Image Models collection.

2. Face Restoration (Output Correction)

Facial Feature Correction

Powered by CodeFormer / GFPGAN. Automatically detect and repair distorted faces in generated images. Fixes wonky eyes, artifacts, and unnatural textures to create realistic human portraits. Best for portrait photography, group shots, and character design. Try our Face Enhancer free tool for quick fixes.

3. Image Upscaling (Resolution Enhancement)

High-Fidelity Upscaling

Powered by 4x-UltraSharp / RealESRGAN. Increase image resolution by 400% without blurs or pixelation. These models hallucinate realistic details to keep edges sharp at 4K or 8K sizes. Best for print media, marketing materials, and large displays. Try the Image Upscaler or Video Upscaler free tools.

4. ControlNet Preprocessors (Structure Control)

Composition Guidance

Powered by Canny / Depth / OpenPose. Extract structural information (edges, depth maps, or poses) from a reference image to guide the generation process precisely. Best for replicating poses, copying layouts, and architectural rendering. Works seamlessly with FLUX Kontext for precise image editing. Also see Segment Anything for object isolation.

The Optimized Generation Pipeline

How to chain assist tools for the perfect result.

Prompt Refinement

MagicPrompt expands your short keywords into detailed, model-optimized prompts with lighting, style, and composition descriptors.

Image Generation

The refined prompt feeds into your chosen model (FLUX, SDXL, etc.) with optional ControlNet guidance for structural control.

Face Restoration

CodeFormer / GFPGAN automatically detects and repairs any facial distortions in the output. Try our Face Enhancer tool.

Upscaling & Delivery

4x-UltraSharp / RealESRGAN upscales to 4K/8K for print-ready quality. Try our Image Upscaler.

Q & A

What are AI Generation Assist Tools?

These are specialized, smaller AI models designed to support the main generation process. They don't generate images from scratch but rather help prepare the input (prompts) or improve the output (upscaling, fixing).

Do I need to use Prompt Assist?

It is not mandatory, but highly recommended if you want to maximize the potential of models like Stable Diffusion or FLUX. Better prompts almost always lead to better images.

Can CodeFormer fix old photos?

Yes. While primarily used for fixing AI-generated faces, CodeFormer is also excellent at restoring old, blurry, or damaged real-world photographs.

Do assist tools cost extra credits?

Generally, assist tools are much lighter and faster than full generation models, so they consume significantly fewer credits. Some simple pre-processing steps may even be free depending on your plan.

Can I chain these tools automatically?

Yes. With the WaveSpeed API, you can define a "workflow" in your JSON request. For example, you can request generate_image with restore_faces: true and upscale: 2x, and the system will run the assist tools in sequence automatically.