Coercer commited on
Commit
29d0dc1
·
verified ·
1 Parent(s): a01d294

Upload 2 files

Browse files
Training_model_data/config_dataset_sdxl.toml ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [general]
2
+ resolution = 1024
3
+ caption_extension = ".txt"
4
+ shuffle_caption = true
5
+ enable_bucket = true
6
+ min_bucket_reso = 512
7
+ max_bucket_reso = 2048
8
+ bucket_reso_steps = 32
9
+
10
+ [[datasets]]
11
+ keep_tokens = 1
12
+
13
+ [[datasets.subsets]]
14
+ image_dir = "/content/drive/MyDrive/dataset"
15
+ class_tokens = "raichu"
16
+ num_repeats = 3
17
+ keep_tokens = 1
Training_model_data/config_sdxl.toml ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ sdxl = true
4
+ v_parameterization = false
5
+ pretrained_model_name_or_path = "SDXLCheckpoint.safetensors"
6
+
7
+ [optimizer_arguments]
8
+ optimizer_type = "Adafactor"
9
+ optimizer_args = [ "scale_parameter=False", "relative_step=False", "warmup_init=False" ]
10
+ learning_rate = 8e-7
11
+ lr_scheduler = "constant_with_warmup"
12
+ lr_warmup_steps = 200
13
+ max_grad_norm = 0.0
14
+ train_text_encoder = false
15
+ adaptive_noise_scale = 0
16
+ min_snr_gamma = 5
17
+
18
+ [dataset_arguments]
19
+ cache_latents = true
20
+ cache_latents_to_disk = true
21
+ debug_dataset = false
22
+
23
+ [training_arguments]
24
+ save_n_epoch_ratio = 0
25
+ train_batch_size = 1
26
+ mem_eff_attn = false
27
+ max_data_loader_n_workers = 6
28
+ persistent_data_loader_workers = true
29
+ gradient_checkpointing = true
30
+ gradient_accumulation_steps = 1
31
+ mixed_precision = "no"
32
+ save_precision = "fp16"
33
+ logging_dir = "logs"
34
+ log_prefix = "last"
35
+ xformers = true
36
+ sdpa = false
37
+ noise_offset = 0.0357
38
+ seed = 1234
39
+ save_model_as = "safetensors"
40
+ save_state = false
41
+
42
+ [dreambooth_arguments]
43
+ prior_loss_weight = 1.0
44
+
45
+ [sample_prompt_arguments]
46
+
47
+ [saving_arguments]