Gary0076 commited on
Commit
32df9cb
·
1 Parent(s): bd0dc54

feat: upload matsumoto-sarina lora model

Browse files
matsumoto-sarina_config/config_file.toml CHANGED
@@ -5,35 +5,25 @@ pretrained_model_name_or_path = "/content/pretrained_model/Animefull-final-prune
5
 
6
  [additional_network_arguments]
7
  no_metadata = false
8
- unet_lr = 0.00015
9
- text_encoder_lr = 1.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.00015
19
  max_grad_norm = 1.0
20
  lr_scheduler = "constant_with_warmup"
21
  lr_warmup_steps = 5
22
 
23
  [dataset_arguments]
 
24
  debug_dataset = false
25
- in_json = "/content/LoRA/meta_lat.json"
26
- train_data_dir = "/content/LoRA/train_data"
27
- dataset_repeats = 27
28
- shuffle_caption = true
29
- keep_tokens = 0
30
- resolution = "512,512"
31
- caption_dropout_rate = 0
32
- caption_tag_dropout_rate = 0
33
- caption_dropout_every_n_epochs = 0
34
- color_aug = false
35
- token_warmup_min = 1
36
- token_warmup_step = 0
37
 
38
  [training_arguments]
39
  output_dir = "/content/LoRA/output"
@@ -59,5 +49,8 @@ lowram = true
59
  sample_every_n_epochs = 999999
60
  sample_sampler = "ddim"
61
 
 
 
 
62
  [saving_arguments]
63
  save_model_as = "safetensors"
 
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 = 64
12
+ network_alpha = 32
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 = 5
22
 
23
  [dataset_arguments]
24
+ cache_latents = true
25
  debug_dataset = false
26
+ vae_batch_size = 4
 
 
 
 
 
 
 
 
 
 
 
27
 
28
  [training_arguments]
29
  output_dir = "/content/LoRA/output"
 
49
  sample_every_n_epochs = 999999
50
  sample_sampler = "ddim"
51
 
52
+ [dreambooth_arguments]
53
+ prior_loss_weight = 1.0
54
+
55
  [saving_arguments]
56
  save_model_as = "safetensors"
matsumoto-sarina_config/dataset_config.toml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 640
3
+ min_bucket_reso = 256
4
+ max_bucket_reso = 1024
5
+ caption_dropout_rate = 0
6
+ caption_tag_dropout_rate = 0
7
+ caption_dropout_every_n_epochs = 0
8
+ flip_aug = false
9
+ color_aug = false
10
+ [[datasets.subsets]]
11
+ image_dir = "/content/LoRA/train_data"
12
+ class_tokens = "matsumoto_sarina"
13
+ num_repeats = 27
14
+
15
+
16
+ [general]
17
+ enable_bucket = true
18
+ caption_extension = ".txt"
19
+ shuffle_caption = true
20
+ keep_tokens = 1
21
+ bucket_reso_steps = 64
22
+ bucket_no_upscale = false