Gary0076 commited on
Commit
58bd9e7
·
1 Parent(s): 36e9c10

feat: upload sutegoro-shiina lora model

Browse files
sutegoro-shiina_config/config_file.toml CHANGED
@@ -5,21 +5,20 @@ pretrained_model_name_or_path = "/content/pretrained_model/Animefull-final-prune
5
 
6
  [additional_network_arguments]
7
  no_metadata = false
8
- unet_lr = 0.00016
9
- text_encoder_lr = 3e-5
10
  network_module = "networks.lora"
11
  network_dim = 128
12
- network_alpha = 128
13
  network_train_unet_only = false
14
  network_train_text_encoder_only = false
15
 
16
  [optimizer_arguments]
17
  optimizer_type = "AdamW8bit"
18
- learning_rate = 0.00016
19
  max_grad_norm = 1.0
20
- lr_scheduler = "cosine_with_restarts"
21
  lr_warmup_steps = 0
22
- lr_scheduler_num_cycles = 3
23
 
24
  [dataset_arguments]
25
  debug_dataset = false
@@ -28,7 +27,7 @@ debug_dataset = false
28
  output_dir = "/content/LoRA/output"
29
  output_name = "sutegoro-shiina"
30
  save_precision = "fp16"
31
- save_every_n_epochs = 25
32
  train_batch_size = 3
33
  max_token_length = 225
34
  mem_eff_attn = false
@@ -45,7 +44,7 @@ log_prefix = "sutegoro-shiina"
45
  lowram = true
46
 
47
  [sample_prompt_arguments]
48
- sample_every_n_epochs = 999999
49
  sample_sampler = "ddim"
50
 
51
  [saving_arguments]
 
5
 
6
  [additional_network_arguments]
7
  no_metadata = false
8
+ unet_lr = 0.0001
9
+ text_encoder_lr = 5e-5
10
  network_module = "networks.lora"
11
  network_dim = 128
12
+ network_alpha = 64
13
  network_train_unet_only = false
14
  network_train_text_encoder_only = false
15
 
16
  [optimizer_arguments]
17
  optimizer_type = "AdamW8bit"
18
+ learning_rate = 0.0001
19
  max_grad_norm = 1.0
20
+ lr_scheduler = "constant_with_warmup"
21
  lr_warmup_steps = 0
 
22
 
23
  [dataset_arguments]
24
  debug_dataset = false
 
27
  output_dir = "/content/LoRA/output"
28
  output_name = "sutegoro-shiina"
29
  save_precision = "fp16"
30
+ save_every_n_epochs = 5
31
  train_batch_size = 3
32
  max_token_length = 225
33
  mem_eff_attn = false
 
44
  lowram = true
45
 
46
  [sample_prompt_arguments]
47
+ sample_every_n_epochs = 1
48
  sample_sampler = "ddim"
49
 
50
  [saving_arguments]
sutegoro-shiina_config/dataset_config.toml CHANGED
@@ -1,12 +1,12 @@
1
  [[datasets]]
2
- resolution = 768
3
  caption_dropout_rate = 0
4
  caption_tag_dropout_rate = 0
5
  caption_dropout_every_n_epochs = 0
6
  color_aug = false
7
  [[datasets.subsets]]
8
  image_dir = "/content/LoRA/train_data"
9
- num_repeats = 30
10
  metadata_file = "/content/LoRA/meta_lat.json"
11
 
12
 
 
1
  [[datasets]]
2
+ resolution = 512
3
  caption_dropout_rate = 0
4
  caption_tag_dropout_rate = 0
5
  caption_dropout_every_n_epochs = 0
6
  color_aug = false
7
  [[datasets.subsets]]
8
  image_dir = "/content/LoRA/train_data"
9
+ num_repeats = 50
10
  metadata_file = "/content/LoRA/meta_lat.json"
11
 
12