You're looking at a specific version of this model. Jump to the model overview.
edenartlab /sdxl-lora-trainer:f11c84d6
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 |
---|---|---|---|
lora_training_urls |
string
|
Training images for new LORA concept (can be image urls or a .zip file of images)
|
|
concept_mode |
string
|
style
|
'face' / 'style' / 'object' (default)
|
sd_model_version |
string
|
sdxl
|
'sdxl' / 'sd15'
|
name |
string
|
unnamed
|
Name of new LORA concept
|
seed |
integer
|
Random seed for reproducible training. Leave empty to use a random seed
|
|
resolution |
integer
|
512
|
Square pixel resolution which your images will be resized to for training, recommended: 512 or 640
|
train_batch_size |
integer
|
4
|
Batch size (per device) for training
|
max_train_steps |
integer
|
400
|
Number of training steps.
|
token_warmup_steps |
integer
|
0
|
Number of steps for token (textual_inversion) warmup.
|
checkpointing_steps |
integer
|
10000
|
Number of steps between saving checkpoints. Set to very very high number to disable checkpointing, because you don't need intermediate checkpoints.
|
unet_lr |
number
|
0.001
|
final learning rate of unet (after warmup)
|
ti_lr |
number
|
0.001
|
Learning rate for training textual inversion embeddings. Don't alter unless you know what you're doing.
|
freeze_ti_after_completion_f |
number
|
1
|
Fraction of training steps after which to freeze textual inversion embeddings
|
lora_rank |
integer
|
16
|
Rank of LoRA embeddings for the unet.
|
caption_model |
string
|
blip
|
Which captioning model to use. ['gpt4-v', 'blip'] are supported right now
|
n_tokens |
integer
|
2
|
How many new tokens to inject per concept
|
verbose |
boolean
|
True
|
verbose output
|
debug |
boolean
|
False
|
For debugging locally only (dont activate this on replicate)
|
Output schema
The shape of the response you’ll get when you run this model with an API.
Schema
{'items': {'properties': {'attributes': {'title': 'Attributes',
'type': 'object'},
'files': {'default': [],
'items': {'format': 'uri',
'type': 'string'},
'title': 'Files',
'type': 'array'},
'isFinal': {'default': False,
'title': 'Isfinal',
'type': 'boolean'},
'name': {'title': 'Name', 'type': 'string'},
'progress': {'title': 'Progress', 'type': 'number'},
'thumbnails': {'default': [],
'items': {'format': 'uri',
'type': 'string'},
'title': 'Thumbnails',
'type': 'array'}},
'title': 'CogOutput',
'type': 'object'},
'title': 'Output',
'type': 'array',
'x-cog-array-type': 'iterator'}