x1101 commited on
Commit
eb71bd3
·
1 Parent(s): b45c432

feat: upload shirayuki lora model

Browse files
Files changed (1) hide show
  1. shirayuki_config/config_file.toml +6 -7
shirayuki_config/config_file.toml CHANGED
@@ -7,29 +7,28 @@ pretrained_model_name_or_path = "/content/pretrained_model/Animefull-final-prune
7
  no_metadata = false
8
  unet_lr = 0.0005
9
  text_encoder_lr = 0.0001
10
- network_module = "lycoris.kohya"
11
  network_dim = 32
12
  network_alpha = 16
13
- network_args = [ "conv_dim=32", "conv_alpha=16", "algo=lora",]
14
  network_train_unet_only = false
15
  network_train_text_encoder_only = false
16
 
17
  [optimizer_arguments]
 
18
  optimizer_type = "AdamW8bit"
19
  learning_rate = 0.0005
20
  max_grad_norm = 1.0
21
- lr_scheduler = "cosine_with_restarts"
22
  lr_warmup_steps = 0
23
- lr_scheduler_num_cycles = 3
24
 
25
  [dataset_arguments]
26
  debug_dataset = false
27
  in_json = "/content/LoRA/meta_lat.json"
28
  train_data_dir = "/content/LoRA/train_data"
29
- dataset_repeats = 15
30
  shuffle_caption = true
31
  keep_tokens = 0
32
- resolution = "768,768"
33
  caption_dropout_rate = 0
34
  caption_tag_dropout_rate = 0
35
  caption_dropout_every_n_epochs = 0
@@ -59,7 +58,7 @@ lowram = true
59
 
60
  [sample_prompt_arguments]
61
  sample_every_n_epochs = 999999
62
- sample_sampler = "k_dpm_2_a"
63
 
64
  [saving_arguments]
65
  save_model_as = "safetensors"
 
7
  no_metadata = false
8
  unet_lr = 0.0005
9
  text_encoder_lr = 0.0001
10
+ network_module = "networks.lora"
11
  network_dim = 32
12
  network_alpha = 16
 
13
  network_train_unet_only = false
14
  network_train_text_encoder_only = false
15
 
16
  [optimizer_arguments]
17
+ min_snr_gamma = 5
18
  optimizer_type = "AdamW8bit"
19
  learning_rate = 0.0005
20
  max_grad_norm = 1.0
21
+ lr_scheduler = "constant"
22
  lr_warmup_steps = 0
 
23
 
24
  [dataset_arguments]
25
  debug_dataset = false
26
  in_json = "/content/LoRA/meta_lat.json"
27
  train_data_dir = "/content/LoRA/train_data"
28
+ dataset_repeats = 10
29
  shuffle_caption = true
30
  keep_tokens = 0
31
+ resolution = "512,512"
32
  caption_dropout_rate = 0
33
  caption_tag_dropout_rate = 0
34
  caption_dropout_every_n_epochs = 0
 
58
 
59
  [sample_prompt_arguments]
60
  sample_every_n_epochs = 999999
61
+ sample_sampler = "ddim"
62
 
63
  [saving_arguments]
64
  save_model_as = "safetensors"