| pretrained_model_name_or_path = "sd-models/animefull-final-pruned.safetensors" | |
| train_data_dir = "/kaggle/input/lora-train-picture/music_illustration" | |
| shuffle_caption = true | |
| caption_extension = ".txt" | |
| keep_tokens = 1 | |
| resolution = "512,916" | |
| enable_bucket = true | |
| output_dir = "/kaggle/working/output" | |
| output_name = "music_illustration" | |
| save_precision = "fp16" | |
| save_every_n_epochs = 1 | |
| train_batch_size = 2 | |
| max_token_length = 225 | |
| xformers = true | |
| max_train_epochs = 20 | |
| seed = 114514 | |
| gradient_accumulation_steps = 64 | |
| mixed_precision = "fp16" | |
| clip_skip = 2 | |
| logging_dir = "/kaggle/working/output/logs" | |
| log_with = "wandb" | |
| log_prefix = "music_illustration" | |
| log_tracker_name = "music_illustration" | |
| lowram = true | |
| sample_every_n_epochs = 1 | |
| sample_prompts = "/kaggle/input/lora-train-picture/music_illustration/tag.txt" | |
| sample_sampler = "euler_a" | |
| prior_loss_weight = 0.0 | |
| optimizer_type = "Lion" | |
| learning_rate = 7e-5 | |
| optimizer_args = [ "weight_decay=0.01", "betas=.95,.98",] | |
| lr_scheduler = "cosine_with_restarts" | |
| min_snr_gamma = 5.0 | |
| unet_lr = 7e-5 | |
| text_encoder_lr = 8e-6 | |
| network_module = "networks.lora" | |
| network_dim = 64 | |
| network_alpha = 32.0 | |
| training_comment = "music_illustration_sukaka" | |