You're looking at a specific version of this model. Jump to the model overview.
vetkastar /lora-train:d1d54002
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)
|
files/model/juggernautXL_version6Rundiffusion.safetensors
Options: files/model/deliberate_v2.safetensors, files/model/juggernautXL_version6Rundiffusion.safetensors |
Выберите базовую модель Stable Diffusion
|
| dataset_url |
string
|
Архив с данными для обучения
|
|
| network_module |
string
(enum)
|
networks.sdxl
Options: networks.lora, networks.sdxl |
Имя модуля для LoRA
|
| network_dim |
integer
|
128
|
Размерность LoRA
|
| network_alpha |
integer
|
128
|
Коэффициент альфа для LoRA
|
| lora_output_name |
string
|
Lora-XL.safetensors
|
Имя файла обученной LoRA модели
|
| save_safetensors |
boolean
|
True
|
Сохранять ли модель в формате safetensors
|
| max_train_steps |
integer
|
1000
|
Максимальное число шагов обучения
|
| save_every_n_epochs |
integer
|
1
|
Сохранять модель каждые N эпох
|
| learning_rate |
number
|
0.000002
|
Скорость обучения
|
| unet_lr |
number
|
0.0001
|
Скорость обучения U-Net
|
| text_encoder_lr |
number
|
0.00005
|
Скорость обучения Text Encoder
|
| lr_scheduler |
string
|
constant_with_warmup
|
Тип планировщика скорости обучения
|
| lr_warmup |
integer
|
10
|
Количество шагов разогрева скорости обучения
|
| optimizer_type |
string
|
AdamW8bit
|
Тип оптимизатора (AdamW8bit, AdamW, Lion)
|
| train_batch_size |
integer
|
8
|
Размер батча для обучения
|
| max_train_epochs |
integer
|
1
|
Количество эпох обучения
|
| gradient_accumulation_steps |
integer
|
1
|
Количество шагов аккумуляции градиента
|
| mixed_precision |
string
|
bf16
|
Точность вычислений (bf16, fp16)
|
| save_precision |
string
|
bf16
|
Точность сохранения модели (bf16, fp16)
|
| clip_skip |
integer
|
1
|
Количество слоев CLIP для пропуска
|
| max_token_length |
integer
|
150
|
Максимальная длина токена
|
| seed |
integer
|
1
|
Начальное значение для генератора случайных чисел
|
| prior_loss_weight |
number
|
1
|
Вес функции потерь Prior
|
| caption_dropout_rate |
number
|
0
|
Вероятность dropout для текстовых описаний
|
| caption_dropout_every_n_epochs |
integer
|
0
|
Применять dropout каждые N эпох
|
| max_data_loader_n_workers |
integer
|
0
|
Максимальное количество процессов загрузки данных
|
| persistent_data_loader_workers |
boolean
|
False
|
Использовать постоянные процессы загрузки данных
|
| bucket_no_upscale |
boolean
|
False
|
Не увеличивать разрешение при batching
|
| bucket_reso_steps |
integer
|
32
|
Шаги разрешения при batching
|
| keep_tokens |
integer
|
1
|
Количество токенов для сохранения
|
| color_aug |
boolean
|
False
|
Использовать цветовую аугментацию
|
| flip_aug |
boolean
|
False
|
Использовать аугментацию отражением
|
| gradient_checkpointing |
boolean
|
True
|
Использовать gradient checkpointing
|
| shuffle_caption |
boolean
|
False
|
Перемешивать текстовые описания
|
| full_fp16 |
boolean
|
False
|
Использовать полное FP16 для обучения
|
| full_bf16 |
boolean
|
False
|
Использовать полное BF16 для обучения
|
| xformers |
boolean
|
True
|
Использовать xformers
|
| save_model_as |
string
|
safetensors
|
Формат сохранения модели (safetensors, ckpt, pt)
|
| save_state |
boolean
|
False
|
Сохранять состояние модели
|
| sdxl |
boolean
|
True
|
Обучать модель SDXL
|
| scale_weight_norms |
number
|
0
|
Масштабировать нормы весов
|
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'}