Views
No views yet
1[network_arguments]
2network_dim = 32
3network_alpha = 32
4network_module = "networks.lora_flux_2"
5
6[optimizer_arguments]
7learning_rate = 1.0
8optimizer_type = "prodigyplus.prodigy_plus_schedulefree.ProdigyPlusScheduleFree"
9optimizer_args = ["betas=(0.9, 0.999)", "use_speed=True", "use_orthograd=True", "use_bias_correction=True"]
10
11[training_arguments]
12max_train_epochs = 20 # should be about 10k~ steps.
13seed = 42
14timestep_sampling = "custom_flux2"
15weighting_scheme = "none"
16mixed_precision = "bf16"
17flash_attn = true
18gradient_checkpointing = true
19compile = true
20max_data_loader_n_workers = 4
21persistent_data_loader_workers = true
22
23[sampling]
24sample_every_n_epochs = 1
25sample_prompts = "anime-stripper-klein9b-v4/prompts.txt"
26sample_at_first = true
27
28[saving_arguments]
29save_every_n_epochs = 1
30wandb_api_key = ""
31output_dir = "anime-stripper-klein9b-v4/output"
32output_name = "anime-stripper-klein9b-v4"
33wandb_run_name = "anime-stripper-klein9b-v4"
34log_prefix = "anime-stripper-klein9b-v4"
35log_with = "wandb"1[general]
2caption_extension = ".txt"
3enable_bucket = true
4bucket_no_upscale = false
5batch_size = 8
6
7[[datasets]]
8resolution = 1024
9num_repeats = 1
10image_jsonl_file = "anime-stripper-klein9b-v4/dataset.jsonl"
11cache_directory = "anime-stripper-klein9b-v4/datasetb"