pretrained_model_name_or_path = "sd-models/animefull-final-pruned.safetensors" | |
train_data_dir = "/kaggle/input/lora-train-picture/Portrait_of_a_girl/Portrait_of_a_girl" | |
shuffle_caption = true | |
caption_extension = ".txt" | |
keep_tokens = 5 | |
resolution = "512,512" | |
enable_bucket = true | |
output_dir = "/kaggle/working/output" | |
output_name = "Portrait_of_a_girl" | |
save_precision = "fp16" | |
save_every_n_epochs = 1 | |
train_batch_size = 2 | |
max_token_length = 225 | |
xformers = true | |
max_train_epochs = 8 | |
seed = 1337 | |
gradient_accumulation_steps = 64 | |
mixed_precision = "fp16" | |
clip_skip = 2 | |
logging_dir = "/kaggle/working/logs" | |
log_with = "wandb" | |
log_prefix = "Portrait_of_a_girl" | |
log_tracker_name = "Portrait_of_a_girl" | |
multires_noise_iterations = 6 | |
multires_noise_discount = 0.1 | |
lowram = true | |
sample_every_n_epochs = 1 | |
sample_prompts = "/kaggle/input/lora-train-picture/Portrait_of_a_girl/test-picture.txt" | |
sample_sampler = "euler_a" | |
optimizer_type = "DAdaptation" | |
learning_rate = 1.0 | |
optimizer_args = [ "decouple=True", "weight_decay=0.01",] | |
lr_scheduler = "cosine_with_restarts" | |
min_snr_gamma = 5.0 | |
unet_lr = 1.0 | |
text_encoder_lr = 1.0 | |
network_module = "networks.lora" | |
network_dim = 128 | |
network_alpha = 64.0 | |
training_comment = "Portrait_of_a_girl-sukaka" | |