up

Nsfw-Full_config/config_file.toml

@@ -1,31 +1,32 @@
 [model_arguments]
 v2 = false
 v_parameterization = false
-pretrained_model_name_or_path = "/content/pretrained_model/Anything-v3-1.safetensors"
+pretrained_model_name_or_path = "/content/pretrained_model/anylora_novaeFp16Pruned.ckpt"
 vae = "/content/vae/anime.vae.pt"
 
 [additional_network_arguments]
 no_metadata = false
-unet_lr = 0.0001
-text_encoder_lr = 5e-5
+unet_lr = 0.0005
+text_encoder_lr = 0.0001
 network_module = "networks.lora"
-network_dim = 128
-network_alpha = 128
+network_dim = 16
+network_alpha = 8
 network_train_unet_only = false
 network_train_text_encoder_only = false
 
 [optimizer_arguments]
 optimizer_type = "AdamW8bit"
-learning_rate = 0.0001
+learning_rate = 0.0005
 max_grad_norm = 1.0
-lr_scheduler = "constant"
+lr_scheduler = "cosine_with_restarts"
 lr_warmup_steps = 0
+lr_scheduler_num_cycles = 3
 
 [dataset_arguments]
 debug_dataset = false
 
 [training_arguments]
-output_dir = "/content/drive/MyDrive/LoRA/output"
+output_dir = "/content/LoRA/output"
 output_name = "Nsfw-Full"
 save_precision = "fp16"
 save_every_n_epochs = 3