lora / styles /agm /auto_save_agm.toml
derriandistro's picture
Upload 81 files
eb8a567
raw
history blame
1.82 kB
[[subsets]]
num_repeats = 2
keep_tokens = 0
caption_extension = ".txt"
shuffle_caption = true
flip_aug = false
color_aug = false
random_crop = true
is_reg = false
image_dir = "F:/gallery-dl/gallery-dl/pixiv/agm"
caption_dropout_rate = 0.04
caption_dropout_every_n_epochs = 0
caption_tag_dropout_rate = 0.0
[sample_args]
[logging_args]
[general_args.args]
pretrained_model_name_or_path = "F:/Desktop/stable diffusion/LoRA/AnimeFullFinal.safetensors"
mixed_precision = "bf16"
seed = 23
clip_skip = 2
max_data_loader_n_workers = 1
persistent_data_loader_workers = true
max_token_length = 225
prior_loss_weight = 1.0
xformers = true
max_train_epochs = 30
[general_args.dataset_args]
resolution = 896
batch_size = 8
[network_args.args]
network_dim = 8
network_alpha = 1.0
min_timestep = 0
max_timestep = 1000
network_dropout = 0.5
[optimizer_args.args]
optimizer_type = "AdamW8bit"
lr_scheduler = "cosine"
learning_rate = 0.0001
max_grad_norm = 1.0
lr_scheduler_type = "LoraEasyCustomOptimizer.CustomOptimizers.CosineAnnealingWarmupRestarts"
lr_scheduler_num_cycles = 4
unet_lr = 0.0005
warmup_ratio = 0.1
min_snr_gamma = 8
scale_weight_norms = 5.0
[saving_args.args]
output_dir = "F:/stable_diffusion_models_and_outputs/models/Lora/v1/Styles/agm"
save_precision = "fp16"
save_model_as = "safetensors"
output_name = "agm"
save_every_n_epochs = 2
tag_occurrence = true
save_toml = true
[bucket_args.dataset_args]
enable_bucket = true
min_bucket_reso = 512
max_bucket_reso = 1536
bucket_reso_steps = 64
[noise_args.args]
multires_noise_iterations = 10
multires_noise_discount = 0.35
[network_args.args.network_args]
module_dropout = 0.25
conv_dim = 24
conv_alpha = 12.0
[optimizer_args.args.lr_scheduler_args]
min_lr = 1e-6
gamma = 0.85
[optimizer_args.args.optimizer_args]
weight_decay = "0.1"
betas = "0.9,0.99"