You're looking at a specific version of this model. Jump to the model overview.
prunaai /p-image-trainer:e6b9761c
Input schema
The fields you can use to run this model with an API. If you don’t give a value for a field its default value will be used.
| Field | Type | Default value | Description |
|---|---|---|---|
| training_type |
None
|
balanced
|
Type of training to perform
|
| steps |
integer
|
1000
Min: 100 Max: 5000 |
Total number of training steps
|
| image_data |
string
|
Zip archive with images of a consistent style. Use at least 10 images. The zip can also contain a text file for each image (e.g., photo.txt for photo.jpg) to specify edit instructions for the image pair.
|
|
| learning_rate |
number
|
0.0001
Min: 0.00001 Max: 0.01 |
Learning rate applied to trainable parameters
|
| default_caption |
string
|
Default caption to use when caption files are missing. If not provided and captions are missing, training will fail.
|
Output schema
The shape of the response you’ll get when you run this model with an API.
Schema
{'format': 'uri', 'title': 'Output', 'type': 'string'}