feat: upload asahi lora model

asahi_config/config_file.toml

@@ -6,7 +6,7 @@ pretrained_model_name_or_path = "/content/pretrained_model/Animefull-final-prune
 [additional_network_arguments]
 no_metadata = false
 unet_lr = 1.0
-text_encoder_lr = 0.5
+text_encoder_lr = 1.0
 network_module = "networks.lora"
 network_dim = 32
 network_alpha = 32
@@ -14,10 +14,10 @@ network_train_unet_only = false
 network_train_text_encoder_only = false
 
 [optimizer_arguments]
-optimizer_type = "DAdaptAdam"
+optimizer_type = "Prodigy"
 learning_rate = 1.0
 max_grad_norm = 1.0
-optimizer_args = [ "decouple=True", "weight_decay=0.01", "betas=0.9,0.99",]
+optimizer_args = [ "decouple=True", "weight_decay=0.01", "d_coef=2", "use_bias_correction=True", "safeguard_warmup=True", "betas=0.9,0.99",]
 lr_scheduler = "constant"
 lr_warmup_steps = 0
 
@@ -30,12 +30,12 @@ vae_batch_size = 4
 output_dir = "/content/LoRA/output"
 output_name = "asahi"
 save_precision = "fp16"
-save_every_n_epochs = 2
+save_every_n_epochs = 10
 train_batch_size = 3
 max_token_length = 225
 mem_eff_attn = false
 xformers = true
-max_train_epochs = 10
+max_train_epochs = 100
 max_data_loader_n_workers = 8
 persistent_data_loader_workers = true
 seed = 31337
@@ -48,7 +48,7 @@ log_prefix = "asahi"
 lowram = true
 
 [sample_prompt_arguments]
-sample_every_n_epochs = 1
+sample_every_n_epochs = 10
 sample_sampler = "ddim"
 
 [dreambooth_arguments]

asahi_config/dataset_config.toml

@@ -10,7 +10,7 @@ color_aug = false
 [[datasets.subsets]]
 image_dir = "/content/LoRA/train_data"
 class_tokens = "mksks"
-num_repeats = 10
+num_repeats = 2
 
 
 [general]