[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/10_Ogipote_bokeh_line_drop"
keep_tokens = 5
name = "10_Ogipote_bokeh_line_drop"
num_repeats = 10
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/10_Watanabe_bamboo"
keep_tokens = 5
name = "10_Watanabe_bamboo"
num_repeats = 10
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/10_Watanabe_close_up"
keep_tokens = 5
name = "10_Watanabe_close_up"
num_repeats = 10
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/10_Watanabe_eye_focus"
keep_tokens = 5
name = "10_Watanabe_eye_focus"
num_repeats = 10
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/12_Ogipote_line_drop_scenery"
keep_tokens = 5
name = "12_Ogipote_line_drop_scenery"
num_repeats = 12
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/15_bunoshii_shinobu_best"
keep_tokens = 5
name = "15_bunoshii_shinobu_best"
num_repeats = 15
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/15_Watanabe_school_desk_balancing_not_watanabe"
keep_tokens = 5
name = "15_Watanabe_school_desk_balancing_not_watanabe"
num_repeats = 15
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/15_Watanabe_school_desk_pile"
keep_tokens = 5
name = "15_Watanabe_school_desk_pile"
num_repeats = 15
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/15_Watanabe_school_desk_pile_not_watanabe"
keep_tokens = 5
name = "15_Watanabe_school_desk_pile_not_watanabe"
num_repeats = 15
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/15_Watanabe_tongue"
keep_tokens = 5
name = "15_Watanabe_tongue"
num_repeats = 15
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/1_Watanabe_anothers"
keep_tokens = 5
name = "1_Watanabe_anothers"
num_repeats = 1
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/1_Watanabe_grisaia"
keep_tokens = 5
name = "1_Watanabe_grisaia"
num_repeats = 1
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/1_watanabe_monogatari"
keep_tokens = 5
name = "1_watanabe_monogatari"
num_repeats = 1
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/1_Watanabe_retro_artstyle"
keep_tokens = 5
name = "1_Watanabe_retro_artstyle"
num_repeats = 1
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/1_Watanabe_scan"
keep_tokens = 5
name = "1_Watanabe_scan"
num_repeats = 1
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/1_Watanabe_selective_colorization_not_watanabe"
keep_tokens = 5
name = "1_Watanabe_selective_colorization_not_watanabe"
num_repeats = 1
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/20_bunoshii_shinobu_close_up"
keep_tokens = 5
name = "20_bunoshii_shinobu_close_up"
num_repeats = 20
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/20_Ogipote_close-up"
keep_tokens = 5
name = "20_Ogipote_close-up"
num_repeats = 20
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/20_Ogipote_close-up_shinobu"
keep_tokens = 5
name = "20_Ogipote_close-up_shinobu"
num_repeats = 20
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/20_Ogipote_simple_bokeh"
keep_tokens = 5
name = "20_Ogipote_simple_bokeh"
num_repeats = 20
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/20_Watanabe_industrial"
keep_tokens = 5
name = "20_Watanabe_industrial"
num_repeats = 20
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/20_Watanabe_scaffolding"
keep_tokens = 5
name = "20_Watanabe_scaffolding"
num_repeats = 20
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/25_watanabe_sky"
keep_tokens = 5
name = "25_watanabe_sky"
num_repeats = 25
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/2_Ogipote_not_shinobu_tags"
keep_tokens = 5
name = "2_Ogipote_not_shinobu_tags"
num_repeats = 2
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/2_Watanabe_kizumonogatari"
keep_tokens = 5
name = "2_Watanabe_kizumonogatari"
num_repeats = 2
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/2_watanabe_monogatari_better"
keep_tokens = 5
name = "2_watanabe_monogatari_better"
num_repeats = 2
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/2_Watanabe_shinobu_kizu_small"
keep_tokens = 5
name = "2_Watanabe_shinobu_kizu_small"
num_repeats = 2
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/2_Watanabe_white_obj_screencap"
keep_tokens = 5
name = "2_Watanabe_white_obj_screencap"
num_repeats = 2
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/30_bunoshii_chair"
keep_tokens = 5
name = "30_bunoshii_chair"
num_repeats = 30
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/30_Watanabe_backl"
keep_tokens = 5
name = "30_Watanabe_backl"
num_repeats = 30
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/35_Ogipote_close-up_one_eye"
keep_tokens = 5
name = "35_Ogipote_close-up_one_eye"
num_repeats = 35
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/3_bunoshii"
keep_tokens = 5
name = "3_bunoshii"
num_repeats = 3
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/3_bunoshii_naoetsu_uniform"
keep_tokens = 5
name = "3_bunoshii_naoetsu_uniform"
num_repeats = 3
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/3_bunoshii_other_complete"
keep_tokens = 5
name = "3_bunoshii_other_complete"
num_repeats = 3

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/3_bunoshii_shinobus_dress"
keep_tokens = 5
name = "3_bunoshii_shinobus_dress"
num_repeats = 3
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/3_watanabe_apartments"
keep_tokens = 5
name = "3_watanabe_apartments"
num_repeats = 30
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/40_bunoshii_eye_focus"
keep_tokens = 5
name = "40_bunoshii_eye_focus"
num_repeats = 40
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/40_Ogipote_eye_focus"
keep_tokens = 5
name = "40_Ogipote_eye_focus"
num_repeats = 40
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/4_Ogipote_line_drops"
keep_tokens = 5
name = "4_Ogipote_line_drops"
num_repeats = 4
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/4_Watanabe_monochrome_style"
keep_tokens = 5
name = "4_Watanabe_monochrome_style"
num_repeats = 4
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/4_Watanabe_shinobus_dress"
keep_tokens = 5
name = "4_Watanabe_shinobus_dress"
num_repeats = 4
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/4_Watanabe_shinobus_pink"
keep_tokens = 5
name = "4_Watanabe_shinobus_pink"
num_repeats = 4
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/4_Watanabe_shinobu_complete"
keep_tokens = 5
name = "4_Watanabe_shinobu_complete"
num_repeats = 4
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/5_Watanabe_close-up_anothers"
keep_tokens = 5
name = "5_Watanabe_close-up_anothers"
num_repeats = 5
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/5_Watanabe_foreground"
keep_tokens = 5
name = "5_Watanabe_foreground"
num_repeats = 5
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/5_Watanabe_shade"
keep_tokens = 5
name = "5_Watanabe_shade"
num_repeats = 5
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/5_Watanabe_shade_screencap"
keep_tokens = 5
name = "5_Watanabe_shade_screencap"
num_repeats = 5
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/5_Watanabe_white_obj"
keep_tokens = 5
name = "5_Watanabe_white_obj"
num_repeats = 5
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/60_Watanabe_shaft_tilt"
keep_tokens = 5
name = "60_Watanabe_shaft_tilt"
num_repeats = 60
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/60_Watanabe_sjh_pose"
keep_tokens = 5
name = "60_Watanabe_sjh_pose"
num_repeats = 60
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/6_Ogipote_foreground_line_drops"
keep_tokens = 5
name = "6_Ogipote_foreground_line_drops"
num_repeats = 6
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/6_Ogipote_shinobus_dress"
keep_tokens = 5
name = "6_Ogipote_shinobus_dress"
num_repeats = 6
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/6_Ogipote_shinobu_short_hair"
keep_tokens = 5
name = "6_Ogipote_shinobu_short_hair"
num_repeats = 6
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/6_Ogipote_shinobu_tags"
keep_tokens = 5
name = "6_Ogipote_shinobu_tags"
num_repeats = 6
shuffle_caption = true

[[subsets]]
caption_extension = ".txt"
image_dir = "C:/full_dataset/6_Ogipote_simple_blurry_foreground"
keep_tokens = 5
name = "6_Ogipote_simple_blurry_foreground"
num_repeats = 6
shuffle_caption = true

[train_mode]
train_mode = "lora"

[general_args.args]
persistent_data_loader_workers = true
mixed_precision = "bf16"
gradient_checkpointing = true
gradient_accumulation_steps = 2
seed = 23
max_data_loader_n_workers = 1
max_token_length = 225
prior_loss_weight = 1.0
xformers = true
cache_latents = true
cache_latents_to_disk = true
sdxl = true
v_parameterization = true
pretrained_model_name_or_path = "C:/Users/Admin/Downloads/noobaiXLNAIXL_vPred10Version.safetensors"
max_train_epochs = 8

[general_args.dataset_args]
resolution = 1024
batch_size = 15

[network_args.args]
network_dim = 32
network_alpha = 32.0
min_timestep = 0
max_timestep = 1000

[optimizer_args.args]
optimizer_type = "ProdigyPlusScheduleFree"
lr_scheduler = "constant"
loss_type = "smooth_l1"
huber_c = 0.1
huber_schedule = "snr"
learning_rate = 1.0
unet_lr = 1.0
text_encoder_lr = 1.0
max_grad_norm = 0.1
zero_terminal_snr = true

[saving_args.args]
save_precision = "bf16"
save_model_as = "safetensors"
save_every_n_steps = 500
save_toml = true
save_toml_location = "C:/Users/Admin/Downloads/output_model"
output_dir = "C:/Users/Admin/Downloads/noob_test"
output_name = "Noob_vpred_test"

[logging_args.args]
log_prefix_mode = "output_name"
run_name_mode = "default"
log_with = "tensorboard"
logging_dir = "C:/Users/Admin/Downloads/noob_test/graph"

[edm_loss_args.args]
edm2_loss_weighting = true
edm2_loss_weighting_optimizer = "LoraEasyCustomOptimizer.fmarscrop.FMARSCropV2ExMachina"
edm2_loss_weighting_optimizer_lr = "5e-3"
edm2_loss_weighting_optimizer_args = "{'update_strategy':'cautious', 'gamma':0.0, 'betas':(0.99,0.9999,0.999), 'adaptive_clip':0}"
edm2_loss_weighting_max_grad_norm = "0"
edm2_loss_weighting_generate_graph = true
edm2_loss_weighting_generate_graph_every_x_steps = 10
edm2_loss_weighting_generate_graph_y_limit = 5
edm2_loss_weighting_initial_weights = ""
edm2_loss_weighting_num_channels = 448
edm2_loss_weighting_generate_graph_output_dir = "C:/Users/Admin/Downloads/noob_test"

[extra_args.args]
validation_seed = "1337"
validation_every_n_step = "10"
max_validation_steps = "40"
validation_timesteps = "[50, 200, 400, 600, 800, 950]"

[bucket_args.dataset_args]
enable_bucket = true
bucket_no_upscale = true
min_bucket_reso = 512
max_bucket_reso = 2048
bucket_reso_steps = 64

[network_args.args.network_args]
conv_dim = 32
conv_alpha = 32.0
algo = "full"
train_norm = true
dropout = 0.15

[optimizer_args.args.optimizer_args]
betas = "0.9,0.1"