You're looking at a specific version of this model. Jump to the model overview.
vetkastar /lora-train:f17099c4
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 |
|---|---|---|---|
| images |
string
|
Zip archive containing training images
|
|
| num_epochs |
integer
|
1
|
Number of training epochs
|
| learning_rate |
number
|
0.0001
|
Learning rate for LoRA training
|
| lr_scheduler |
string
(enum)
|
constant
Options: linear, cosine, cosine_with_restarts, polynomial, constant, constant_with_warmup |
Learning rate scheduler
|
| lr_warmup_steps |
integer
|
0
|
Number of warmup steps for learning rate scheduler
|
| output_dir |
string
|
lora_output
|
Output directory for trained LoRA model
|
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'}