You're looking at a specific version of this model. Jump to the model overview.
vetkastar /lora-train:d3b96972
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
|
networks.lora
|
Имя модуля для 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
|
Размер батча для обучения
|
| epoch |
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
|
Максимальная длина токена
|
| max_resolution |
string
|
1024,1024
|
Максимальное разрешение изображения
|
| stop_text_encoder_training |
integer
|
0
|
Шаг, на котором прекращается обучение Text Encoder
|
| seed |
integer
|
1
|
Начальное значение для генератора случайных чисел
|
| prior_loss_weight |
number
|
1
|
Вес функции потерь Prior
|
| caption_dropout_rate |
number
|
0
|
Вероятность dropout для текстовых описаний
|
| caption_dropout_every_n_epochs |
integer
|
0
|
Применять dropout каждые N эпох
|
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'}