You're looking at a specific version of this model. Jump to the model overview.

vetkastar /lora-train:d77bc6de

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
75
Максимальная длина токена
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
number
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'}