SandBase is live — $1 in free credits on signupStart free ›
Use in agentBaidu models

Baidu modelsimage generation api

baidu/ernie-image-trainer

Ernie Image Trainer by Baidu - advanced AI model for training. Delivers high-quality results with fast inference, suitable for both creative and production workflows.

Input
URL to the input data zip archive. The zip should contain pairs of images and corresponding captions. The images should be named: ROOT.EXT. For example: 001.jpg The corresponding captions should be named: ROOT.txt. For example: 001.txt If no text file is provided for an image, the default_caption will be used.
Default caption to use when caption files are missing. If None, missing captions will cause an error.
Learning rate.
1040000
Number of steps to train for Range: 10 to 40000.
Idle

Example output — click Run to generate your own

API README

ERNIE-Image Trainer

baidu/ernie-image-trainer is a dedicated custom image-model training route in Baidu's ERNIE image family. It is designed around the named transformation, using the supplied text or media references to produce a coherent creative result rather than exposing a generic endpoint with loosely related controls.

The route combines its task-specific conditioning with the family's strengths in knowledge-aware image synthesis, Chinese-language understanding, typography, and detailed composition. This makes it useful for production workflows that need deliberate art direction, recognizable subjects, consistent scene logic, and a finished asset that can move directly into review or downstream editing.

Highlights

Subject adaptation. Learns a reusable visual concept from a focused set of training images.

Identity consistency. Encodes recurring subject features for later generation across new scenes and compositions.

Promptable reuse. Turns the trained concept into a controllable element for downstream image creation.

Style specialization. Supports adaptation toward a distinctive aesthetic or production-specific visual language.

Pricing

Billing unitPrice
Per request$1.2

When to Use

✅ Good fit❌ Consider alternatives
The model's named workflow matches the source material and intended outputA different input modality or model route is required
A managed asynchronous result is suitable for the production pipelineA synchronous, interactive editor is essential
The documented controls cover the required duration, framing, or formatThe project needs controls outside this endpoint's schema
Creative iteration benefits from a repeatable request structureExact deterministic pixels, frames, geometry, or samples are mandatory
A finished downloadable media asset is the desired deliverableEditable source layers or a native project file are required

Prompt Guide

For generation, state the intended result first, then add the subject or source treatment, progression, style, and delivery constraints. Keep one creative variable per phrase, use the documented field names for controls, and change one setting at a time when comparing results.

{}

Technical Specs

SpecValue
Model IDbaidu/ernie-image-trainer
Inputsdefault_caption, images_data_url, learning_rate, steps
Required inputsNone
Output fieldscontent_type, url
ExecutionAsync (submit, then poll for result)

Related Models

Related Models