Pruna Ai P Image Text To Image Trainer
Playground
Try it on WavespeedAI!Pruna AI P-Image Text to Image Trainer is a fast AI model training workflow for customizing text-to-image generation models with user-provided data. Ready-to-use REST inference API for training custom styles, brand-specific visuals, character concepts, product image generation, marketing creatives, and personalized AI image workflows with simple integration, no coldstarts, and affordable pricing.
Features
Pruna AI P-Image Text-to-Image Trainer
Pruna AI P-Image Text-to-Image Trainer is a fast training workflow for creating custom LoRAs for the Pruna text-to-image ecosystem. Upload your training image data, optionally provide a default caption, choose the number of training steps, and generate a LoRA that can be used to steer future image generation toward your style, subject, or brand look.
Why Choose This?
-
Fast custom LoRA training Train a text-to-image LoRA for specialized styles, subjects, or branded visuals.
-
Simple training interface Provide your training image data, optional default caption, and training steps without a complex setup process.
-
Optional caption guidance Use
default_captionto provide consistent text conditioning across the training dataset. -
Flexible training depth Use
stepsto balance speed, cost, and how strongly the LoRA learns your dataset. -
Built for the Pruna image stack Trained outputs are intended for downstream use with Pruna text-to-image LoRA workflows.
-
Production-ready API Suitable for custom style pipelines, branded asset generation, and repeatable image workflow customization.
Parameters
| Parameter | Required | Description |
|---|---|---|
| image_data | Yes | Training image data used to create the LoRA. |
| default_caption | No | Optional default caption applied across the training workflow for more consistent conditioning. |
| steps | No | Number of training steps. Higher values generally increase training time and cost. Default: 101. |
How to Use
- Upload your training data — provide the image dataset you want to use for training.
- Add a default caption (optional) — use a short caption if you want more consistent text conditioning across the dataset.
- Set training steps — choose how many steps to run based on your desired balance of speed and training strength.
- Submit — start the training job.
- Use the trained LoRA — apply the resulting LoRA in downstream Pruna text-to-image LoRA workflows.
Example Workflow
Train a custom style LoRA from a curated image set, optionally using a shared default caption, then use the resulting weights in Pruna AI P-Image Text-to-Image LoRA for generation.
Pricing
Pricing is based on the selected steps value.
| Steps | Cost |
|---|---|
| 100 | $0.18 |
| 101 | $0.1818 |
| 250 | $0.45 |
| 500 | $0.90 |
| 1000 | $1.80 |
| 2000 | $3.60 |
Billing Rules
- Pricing scales linearly with
steps - Cost is $1.80 per 1,000 steps
- Higher
stepsvalues increase total training cost proportionally default_captiondoes not affect pricing
Best Use Cases
- Custom style training — Create LoRAs for a distinct visual style or art direction.
- Brand consistency — Train reusable LoRAs for campaigns, products, or branded aesthetics.
- Subject-focused generation — Teach the model a recurring character, fashion look, or visual concept.
- Creative workflow personalization — Build specialized LoRAs for repeatable prompt-driven generation.
- Marketing asset pipelines — Create tailored generation tools for ongoing content production.
Pro Tips
- Use a clean, consistent training dataset for better LoRA quality.
- Add a
default_captionwhen your dataset shares a common concept, subject, or style cue. - Start with a moderate number of steps before pushing to larger training runs.
- Increase
stepsgradually if the initial LoRA is too weak or underfit. - Keep your dataset focused on the style or subject you want the LoRA to learn.
- Test the trained LoRA in downstream generation workflows before scaling up training volume.
Notes
image_datais required.default_captionis optional.stepsdefaults to101.- Pricing depends only on the selected
stepsvalue. - LoRAs trained here are intended for Pruna text-to-image LoRA usage.
Related Models
- Pruna AI P-Image Text-to-Image LoRA — Generate images from prompts using a trained LoRA.
- Pruna AI P-Image Text-to-Image — Generate images directly from prompts without a LoRA.
- Pruna AI P-Image Edit Trainer — Train LoRAs for downstream image editing workflows.
- Pruna AI P-Image Edit LoRA — Edit images with LoRA-guided control.
Authentication
For authentication details, please refer to the Authentication Guide.
API Endpoints
Submit Task & Query Result
# Submit the task
curl --location --request POST "https://api.wavespeed.ai/api/v3/pruna-ai/p-image/text-to-image-trainer" \
--header "Content-Type: application/json" \
--header "Authorization: Bearer ${WAVESPEED_API_KEY}" \
--data-raw '{
"steps": 101
}'
# Get the result
curl --location --request GET "https://api.wavespeed.ai/api/v3/predictions/${requestId}/result" \
--header "Authorization: Bearer ${WAVESPEED_API_KEY}"
Parameters
Task Submission Parameters
Request Parameters
| Parameter | Type | Required | Default | Range | Description |
|---|---|---|---|---|---|
| image_data | string | Yes | - | - | Training image archive URL. |
| steps | integer | No | 101 | - | Number of training steps. |
| default_caption | string | No | - | - | Default caption for images without .txt files. If captions are missing and this is empty, training can fail. |
Response Parameters
| Parameter | Type | Description |
|---|---|---|
| code | integer | HTTP status code (e.g., 200 for success) |
| message | string | Status message (e.g., “success”) |
| data.id | string | Unique identifier for the prediction, Task Id |
| data.model | string | Model ID used for the prediction |
| data.outputs | array | Array of URLs to the generated content (empty when status is not completed) |
| data.urls | object | Object containing related API endpoints |
| data.urls.get | string | URL to retrieve the prediction result |
| data.status | string | Status of the task: created, processing, completed, or failed |
| data.created_at | string | ISO timestamp of when the request was created (e.g., “2023-04-01T12:34:56.789Z”) |
| data.error | string | Error message (empty if no error occurred) |
| data.timings | object | Object containing timing details |
| data.timings.inference | integer | Inference time in milliseconds |
Result Request Parameters
| Parameter | Type | Required | Default | Description |
|---|---|---|---|---|
| id | string | Yes | - | Task ID |
Result Response Parameters
| Parameter | Type | Description |
|---|---|---|
| code | integer | HTTP status code (e.g., 200 for success) |
| message | string | Status message (e.g., “success”) |
| data | object | The prediction data object containing all details |
| data.id | string | Unique identifier for the prediction, the ID of the prediction to get |
| data.model | string | Model ID used for the prediction |
| data.outputs | string | Array of URLs to the generated content. |
| data.urls | object | Object containing related API endpoints |
| data.urls.get | string | URL to retrieve the prediction result |
| data.status | string | Status of the task: created, processing, completed, or failed |
| data.created_at | string | ISO timestamp of when the request was created (e.g., “2023-04-01T12:34:56.789Z”) |
| data.error | string | Error message (empty if no error occurred) |
| data.timings | object | Object containing timing details |
| data.timings.inference | integer | Inference time in milliseconds |