Pruna Ai P Image Edit Trainer
Playground
Try it on WavespeedAI!Pruna AI P-Image Edit Trainer is a fast AI model training workflow for customizing image editing models with user-provided data. Ready-to-use REST inference API for training custom edit styles, character-consistent edits, product image updates, brand-specific visuals, marketing assets, and personalized AI image editing workflows with simple integration, no coldstarts, and affordable pricing.
Features
Pruna AI P-Image Edit Trainer
Pruna AI P-Image Edit Trainer is a fast training workflow for creating custom LoRAs for the Pruna image editing stack. Upload your training image data, choose the number of training steps, optionally provide a default caption, and generate a LoRA for downstream edit workflows such as style transfer, character-consistent edits, product edits, and other prompt-guided image editing tasks.
Why Choose This?
-
Fast custom edit LoRA training Train LoRAs specifically for image editing workflows rather than text-to-image generation.
-
Simple training interface Provide training image data and set training steps without a complex setup process.
-
Optional caption guidance Use
default_captionto provide consistent text guidance across the training data. -
Flexible training depth Adjust
stepsto balance speed, cost, and how strongly the LoRA learns your dataset. -
Built for the Pruna edit stack Trained outputs are intended for downstream use with Pruna edit LoRA workflows.
-
Production-ready API Suitable for custom edit styles, character-consistent edits, branded asset pipelines, and repeatable editing workflows.
Parameters
| Parameter | Required | Description |
|---|---|---|
| image_data | Yes | Training image data used to create the edit LoRA. |
| steps | No | Number of training steps. Higher values generally increase training time and cost. Default: 101. |
| default_caption | No | Optional default caption applied to the training workflow for more consistent edit conditioning. |
How to Use
- Upload your training data — provide the image dataset you want to use for training.
- Set training steps — choose how many steps to run based on your desired balance of speed and training strength.
- Add a default caption (optional) — use a short caption if you want more consistent text conditioning during training.
- Submit — start the training job.
- Use the trained LoRA — apply the resulting LoRA in downstream Pruna edit LoRA workflows.
Example Workflow
Train a custom edit LoRA for scene-to-scene style transfer, then use the resulting weights in Pruna AI P-Image Edit LoRA for guided image editing.
Pricing
Pricing is based on the selected steps value.
| Steps | Cost |
|---|---|
| 100 | $0.40 |
| 101 | $0.404 |
| 250 | $1.00 |
| 500 | $2.00 |
| 1000 | $4.00 |
| 2000 | $8.00 |
Billing Rules
- Pricing scales linearly with
steps - Cost is $4.00 per 1,000 steps
- Higher
stepsvalues increase total training cost proportionally default_captiondoes not affect pricing
Best Use Cases
- Custom edit style training — create LoRAs for specific editing aesthetics or transformations
- Character-consistent editing — train reusable LoRAs for recurring character edits
- Product edit workflows — build LoRAs for consistent product transformations and asset updates
- Brand asset editing — create custom edit models for repeatable branded visual workflows
- Personalized image editing — train LoRAs tailored to a specific subject, look, or edit direction
Pro Tips
- Use a clean, focused training dataset for better LoRA quality.
- Start with a moderate number of steps before scaling to larger training runs.
- Use
default_captionwhen you want consistent conditioning across the dataset. - Keep the dataset aligned with the type of edits you want the LoRA to perform later.
- Test the trained LoRA in downstream edit workflows before increasing training volume.
Notes
image_datais required.stepsdefaults to101.default_captionis optional.- Pricing depends only on the selected
stepsvalue. - LoRAs trained here are intended for Pruna edit LoRA usage.
- If you don’t know the format, please see the official file: https://huggingface.co/datasets/davidberenstein1957/enhance/resolve/main/input.zip
Related Models
- Pruna AI P-Image Edit LoRA — Edit images with prompt guidance and a trained edit LoRA.
- Pruna AI P-Image Text-to-Image Trainer — Train LoRAs for text-to-image generation workflows.
- Pruna AI P-Image Text-to-Image LoRA — Generate images from prompts using a trained LoRA.
- Pruna AI P-Image Edit — Edit images with natural-language instructions without a LoRA.
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/edit-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 edit caption for the training images. |
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 |