Gary0076 commited on
Commit
b4536ed
·
1 Parent(s): 5a0a474

feat: upload ultralowleg-bikeshorts lora model

Browse files
ultralowleg-bikeshorts_config/config_file.toml CHANGED
@@ -6,8 +6,8 @@ pretrained_model_name_or_path = "/content/pretrained_model/Animefull-final-prune
6
  [additional_network_arguments]
7
  no_metadata = false
8
  network_module = "networks.lora"
9
- network_dim = 8
10
- network_alpha = 4
11
  network_train_unet_only = false
12
  network_train_text_encoder_only = false
13
 
@@ -15,24 +15,15 @@ network_train_text_encoder_only = false
15
  optimizer_type = "AdamW8bit"
16
  learning_rate = 0.0001
17
  max_grad_norm = 1.0
 
18
  lr_scheduler = "cosine_with_restarts"
19
  lr_warmup_steps = 0
20
  lr_scheduler_num_cycles = 0
21
 
22
  [dataset_arguments]
 
23
  debug_dataset = false
24
- in_json = "/content/LoRA/meta_lat.json"
25
- train_data_dir = "/content/LoRA/train_data"
26
- dataset_repeats = 20
27
- shuffle_caption = true
28
- keep_tokens = 0
29
- resolution = "512,512"
30
- caption_dropout_rate = 0
31
- caption_tag_dropout_rate = 0
32
- caption_dropout_every_n_epochs = 0
33
- color_aug = false
34
- token_warmup_min = 1
35
- token_warmup_step = 0
36
 
37
  [training_arguments]
38
  output_dir = "/content/LoRA/output"
@@ -58,5 +49,8 @@ lowram = false
58
  sample_every_n_epochs = 999999
59
  sample_sampler = "ddim"
60
 
 
 
 
61
  [saving_arguments]
62
  save_model_as = "safetensors"
 
6
  [additional_network_arguments]
7
  no_metadata = false
8
  network_module = "networks.lora"
9
+ network_dim = 16
10
+ network_alpha = 8
11
  network_train_unet_only = false
12
  network_train_text_encoder_only = false
13
 
 
15
  optimizer_type = "AdamW8bit"
16
  learning_rate = 0.0001
17
  max_grad_norm = 1.0
18
+ optimizer_args = [ "weight_decay=0.1",]
19
  lr_scheduler = "cosine_with_restarts"
20
  lr_warmup_steps = 0
21
  lr_scheduler_num_cycles = 0
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"
ultralowleg-bikeshorts_config/dataset_config.toml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 512
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 = ""
13
+ num_repeats = 40
14
+
15
+
16
+ [general]
17
+ enable_bucket = true
18
+ caption_extension = ".txt"
19
+ shuffle_caption = true
20
+ keep_tokens = 0
21
+ bucket_reso_steps = 64
22
+ bucket_no_upscale = false