Skip to content

Commit 7784c97

Browse files
bmaltaisbmaltais
committed
Adding min max reso to other gui
1 parent 7b4798e commit 7784c97

4 files changed

Lines changed: 208 additions & 2 deletions

File tree

‎dreambooth_gui.py‎

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,8 @@ def save_configuration(
7777
full_fp16,
7878
no_token_padding,
7979
stop_text_encoder_training,
80+
min_bucket_reso,
81+
max_bucket_reso,
8082
# use_8bit_adam,
8183
xformers,
8284
save_model_as,
@@ -186,6 +188,8 @@ def open_configuration(
186188
full_fp16,
187189
no_token_padding,
188190
stop_text_encoder_training,
191+
min_bucket_reso,
192+
max_bucket_reso,
189193
# use_8bit_adam,
190194
xformers,
191195
save_model_as,
@@ -294,6 +298,8 @@ def train_model(
294298
full_fp16,
295299
no_token_padding,
296300
stop_text_encoder_training_pct,
301+
min_bucket_reso,
302+
max_bucket_reso,
297303
# use_8bit_adam,
298304
xformers,
299305
save_model_as,
@@ -510,7 +516,7 @@ def train_model(
510516
if v_parameterization:
511517
run_cmd += ' --v_parameterization'
512518
if enable_bucket:
513-
run_cmd += ' --enable_bucket'
519+
run_cmd += f' --enable_bucket --min_bucket_reso={min_bucket_reso} --max_bucket_reso={max_bucket_reso}'
514520
if no_token_padding:
515521
run_cmd += ' --no_token_padding'
516522
if weighted_captions:
@@ -744,6 +750,8 @@ def dreambooth_tab(
744750
advanced_training.full_fp16,
745751
advanced_training.no_token_padding,
746752
basic_training.stop_text_encoder_training,
753+
basic_training.min_bucket_reso,
754+
basic_training.max_bucket_reso,
747755
advanced_training.xformers,
748756
source_model.save_model_as,
749757
advanced_training.shuffle_caption,

‎presets/lora/SDXL - LoRA finetuning phase 1_v1.1.json‎

Lines changed: 95 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,95 @@
1+
{
2+
"LoRA_type": "Standard",
3+
"adaptive_noise_scale": 0,
4+
"additional_parameters": "",
5+
"block_alphas": "",
6+
"block_dims": "",
7+
"block_lr_zero_threshold": "",
8+
"bucket_no_upscale": false,
9+
"bucket_reso_steps": 32,
10+
"cache_latents": true,
11+
"cache_latents_to_disk": true,
12+
"caption_dropout_every_n_epochs": 0.0,
13+
"caption_dropout_rate": 0,
14+
"caption_extension": ".txt2",
15+
"clip_skip": "1",
16+
"color_aug": false,
17+
"conv_alpha": 64,
18+
"conv_alphas": "",
19+
"conv_dim": 64,
20+
"conv_dims": "",
21+
"decompose_both": false,
22+
"dim_from_weights": false,
23+
"down_lr_weight": "",
24+
"enable_bucket": true,
25+
"epoch": 8,
26+
"factor": -1,
27+
"flip_aug": false,
28+
"full_bf16": false,
29+
"full_fp16": false,
30+
"gradient_accumulation_steps": 1.0,
31+
"gradient_checkpointing": true,
32+
"keep_tokens": "0",
33+
"learning_rate": 0.0001,
34+
"lora_network_weights": "",
35+
"lr_scheduler": "constant",
36+
"lr_scheduler_num_cycles": "",
37+
"lr_scheduler_power": "",
38+
"lr_warmup": 0,
39+
"max_bucket_reso": 2048,
40+
"max_data_loader_n_workers": "0",
41+
"max_resolution": "1024,1024",
42+
"max_timestep": 1000,
43+
"max_token_length": "75",
44+
"max_train_epochs": "",
45+
"mem_eff_attn": false,
46+
"mid_lr_weight": "",
47+
"min_bucket_reso": 64,
48+
"min_snr_gamma": 0,
49+
"min_timestep": 0,
50+
"mixed_precision": "bf16",
51+
"module_dropout": 0,
52+
"multires_noise_discount": 0,
53+
"multires_noise_iterations": 0,
54+
"network_alpha": 128,
55+
"network_dim": 128,
56+
"network_dropout": 0,
57+
"no_token_padding": false,
58+
"noise_offset": 0,
59+
"noise_offset_type": "Original",
60+
"num_cpu_threads_per_process": 2,
61+
"optimizer": "AdamW",
62+
"optimizer_args": "",
63+
"persistent_data_loader_workers": false,
64+
"prior_loss_weight": 1.0,
65+
"random_crop": false,
66+
"rank_dropout": 0,
67+
"save_every_n_epochs": 1,
68+
"save_every_n_steps": 0,
69+
"save_last_n_steps": 0,
70+
"save_last_n_steps_state": 1,
71+
"save_precision": "bf16",
72+
"scale_v_pred_loss_like_noise_pred": false,
73+
"scale_weight_norms": 0,
74+
"sdxl": true,
75+
"sdxl_cache_text_encoder_outputs": false,
76+
"sdxl_no_half_vae": true,
77+
"seed": "12345",
78+
"shuffle_caption": false,
79+
"stop_text_encoder_training_pct": 0,
80+
"text_encoder_lr": 0.0001,
81+
"train_batch_size": 8,
82+
"train_on_input": true,
83+
"training_comment": "kill bill, the bride",
84+
"unet_lr": 0.0001,
85+
"unit": 1,
86+
"up_lr_weight": "",
87+
"use_cp": true,
88+
"use_wandb": false,
89+
"v2": false,
90+
"v_parameterization": false,
91+
"vae_batch_size": 0,
92+
"wandb_api_key": "",
93+
"weighted_captions": false,
94+
"xformers": true
95+
}

‎presets/lora/SDXL - LoRA finetuning phase 2_v1.1.json‎

Lines changed: 95 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,95 @@
1+
{
2+
"LoRA_type": "Standard",
3+
"adaptive_noise_scale": 0.00357,
4+
"additional_parameters": "--log_prefix=xl-loha",
5+
"block_alphas": "",
6+
"block_dims": "",
7+
"block_lr_zero_threshold": "",
8+
"bucket_no_upscale": false,
9+
"bucket_reso_steps": 32,
10+
"cache_latents": true,
11+
"cache_latents_to_disk": true,
12+
"caption_dropout_every_n_epochs": 0.0,
13+
"caption_dropout_rate": 0,
14+
"caption_extension": ".txt2",
15+
"clip_skip": "1",
16+
"color_aug": false,
17+
"conv_alpha": 4,
18+
"conv_alphas": "",
19+
"conv_dim": 4,
20+
"conv_dims": "",
21+
"decompose_both": false,
22+
"dim_from_weights": true,
23+
"down_lr_weight": "",
24+
"enable_bucket": true,
25+
"epoch": 1,
26+
"factor": -1,
27+
"flip_aug": false,
28+
"full_bf16": false,
29+
"full_fp16": false,
30+
"gradient_accumulation_steps": 1.0,
31+
"gradient_checkpointing": true,
32+
"keep_tokens": 1,
33+
"learning_rate": 0.0001,
34+
"lora_network_weights": "D:/lycoris/sdxl\\sdxl-kill bill, the bride-lora-1.0av2.safetensors",
35+
"lr_scheduler": "constant",
36+
"lr_scheduler_num_cycles": "1",
37+
"lr_scheduler_power": "",
38+
"lr_warmup": 0,
39+
"max_bucket_reso": 2048,
40+
"max_data_loader_n_workers": "0",
41+
"max_resolution": "1024,1024",
42+
"max_timestep": 1000,
43+
"max_token_length": "75",
44+
"max_train_epochs": "1",
45+
"mem_eff_attn": false,
46+
"mid_lr_weight": "",
47+
"min_bucket_reso": 64,
48+
"min_snr_gamma": 0,
49+
"min_timestep": 0,
50+
"mixed_precision": "bf16",
51+
"module_dropout": 0,
52+
"multires_noise_discount": 0,
53+
"multires_noise_iterations": 0,
54+
"network_alpha": 128,
55+
"network_dim": 128,
56+
"network_dropout": 0,
57+
"no_token_padding": false,
58+
"noise_offset": 0.0357,
59+
"noise_offset_type": "Original",
60+
"num_cpu_threads_per_process": 2,
61+
"optimizer": "AdamW",
62+
"optimizer_args": "",
63+
"persistent_data_loader_workers": false,
64+
"prior_loss_weight": 1.0,
65+
"random_crop": false,
66+
"rank_dropout": 0,
67+
"save_every_n_epochs": 1,
68+
"save_every_n_steps": 0,
69+
"save_last_n_steps": 0,
70+
"save_last_n_steps_state": 0,
71+
"save_precision": "bf16",
72+
"scale_v_pred_loss_like_noise_pred": false,
73+
"scale_weight_norms": 0,
74+
"sdxl": true,
75+
"sdxl_cache_text_encoder_outputs": false,
76+
"sdxl_no_half_vae": true,
77+
"seed": "17415",
78+
"shuffle_caption": false,
79+
"stop_text_encoder_training_pct": 0,
80+
"text_encoder_lr": 0.0001,
81+
"train_batch_size": 1,
82+
"train_on_input": false,
83+
"training_comment": "trigger: portrait",
84+
"unet_lr": 0.0001,
85+
"unit": 1,
86+
"up_lr_weight": "",
87+
"use_cp": false,
88+
"use_wandb": false,
89+
"v2": false,
90+
"v_parameterization": false,
91+
"vae_batch_size": 0,
92+
"wandb_api_key": "",
93+
"weighted_captions": false,
94+
"xformers": true
95+
}

‎textual_inversion_gui.py‎

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,8 @@ def save_configuration(
7777
full_fp16,
7878
no_token_padding,
7979
stop_text_encoder_training,
80+
min_bucket_reso,
81+
max_bucket_reso,
8082
# use_8bit_adam,
8183
xformers,
8284
save_model_as,
@@ -194,6 +196,8 @@ def open_configuration(
194196
full_fp16,
195197
no_token_padding,
196198
stop_text_encoder_training,
199+
min_bucket_reso,
200+
max_bucket_reso,
197201
# use_8bit_adam,
198202
xformers,
199203
save_model_as,
@@ -308,6 +312,8 @@ def train_model(
308312
full_fp16,
309313
no_token_padding,
310314
stop_text_encoder_training_pct,
315+
min_bucket_reso,
316+
max_bucket_reso,
311317
# use_8bit_adam,
312318
xformers,
313319
save_model_as,
@@ -528,7 +534,7 @@ def train_model(
528534
if v_parameterization:
529535
run_cmd += ' --v_parameterization'
530536
if enable_bucket:
531-
run_cmd += ' --enable_bucket'
537+
run_cmd += f' --enable_bucket --min_bucket_reso={min_bucket_reso} --max_bucket_reso={max_bucket_reso}'
532538
if no_token_padding:
533539
run_cmd += ' --no_token_padding'
534540
run_cmd += (
@@ -821,6 +827,8 @@ def ti_tab(
821827
advanced_training.full_fp16,
822828
advanced_training.no_token_padding,
823829
basic_training.stop_text_encoder_training,
830+
basic_training.min_bucket_reso,
831+
basic_training.max_bucket_reso,
824832
advanced_training.xformers,
825833
source_model.save_model_as,
826834
advanced_training.shuffle_caption,

0 commit comments

Comments
 (0)