You're looking at a specific version of this model. Jump to the model overview.
vetkastar /lora-train:46230b14
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 |
---|---|---|---|
pretrained_model_name_or_path |
string
(enum)
|
Options: files/model/deliberate_v2.safetensors, files/model/juggernautXL_version6Rundiffusion.safetensors |
Выберите базовую модель Stable Diffusion
|
dataset_url |
string
|
Архив с данными для обучения
|
|
network_module |
string
|
networks.lora
|
Имя модуля для LoRA
|
network_dim |
integer
|
4
|
Ранг (число измерений) LoRA модели
|
lora_output_name |
string
|
lora_model.safetensors
|
Имя файла обученной LoRA модели
|
save_safetensors |
boolean
|
True
|
Сохранять ли модель в формате safetensors
|
max_train_steps |
integer
|
100
|
Максимальное число шагов обучения
|
learning_rate |
number
|
0.0001
|
Скорость обучения
|
optimizer_type |
string
|
adamw
|
Тип оптимизатора (adam, adamw, adafactor...)
|
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'}