Default38693 commited on
Commit
969245f
·
1 Parent(s): b1583cf

Create AdaFactorBase.toml

Browse files
Files changed (1) hide show
  1. AdaFactor/AdaFactorBase.toml +62 -0
AdaFactor/AdaFactorBase.toml ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Path]
2
+ #pretrained_model_name_or_path = "H:\\WEBUI_Last\\models\\Stable-diffusion\\Trim.safetensors"
3
+ #train_data_dir = "E:\\Dataset\\Akashi(AZ)"
4
+
5
+ #__root_dir = "H:\\sd-scripts\\"
6
+ #logging_dir = "calc:f'{__root_dir}logs'"
7
+ #output_dir = "calc:f'{__root_dir}outputs'"
8
+ #sample_prompts = "calc:f'{__root_dir}prompts\\A.txt'"
9
+
10
+ pretrained_model_name_or_path = "/notebooks/box/Trim.safetensors"
11
+ train_data_dir = "/notebooks/LoRA_Image/"
12
+
13
+ __root_dir = "/notebooks/LoRA/sd-scripts/"
14
+ logging_dir = "calc:f'{__root_dir}logs'"
15
+ output_dir = "calc:f'{__root_dir}outputs'"
16
+ sample_prompts = "/notebooks/LoRA_Setting/A.txt"
17
+
18
+ [Prompt]
19
+ sample_every_n_epochs = 1
20
+ sample_sampler = "k_euler_a"
21
+
22
+ [Else]
23
+ shuffle_caption = true
24
+ caption_extension = ".txt"
25
+ keep_tokens = 1
26
+
27
+ color_aug = true
28
+
29
+ __upper_px = 512
30
+ __resolution_x = "calc:512 + __upper_px"
31
+ __resolution_y = "calc:512 + __upper_px"
32
+
33
+ resolution = "calc:f'{__resolution_x},{__resolution_y}'"
34
+ enable_bucket = true
35
+ min_bucket_reso = "calc:320 + __upper_px"
36
+ bucket_no_upscale = true
37
+ caption_dropout_every_n_epochs = 9
38
+ caption_tag_dropout_rate = 0.2
39
+ save_every_n_epochs = 2
40
+ train_batch_size = 1
41
+ xformers = true
42
+ max_train_epochs = 20
43
+ persistent_data_loader_workers = true
44
+ seed = 987
45
+ mixed_precision = "fp16"
46
+ save_precision = "ref:mixed_precision"
47
+ clip_skip = 2
48
+
49
+ optimizer_type = "AdaFactor"
50
+ optimizer_args = [ "relative_step=True", "scale_parameter=True", "warmup_init=True"]
51
+
52
+ lr_scheduler = "cosine_with_restarts"
53
+ lr_warmup_steps = 500
54
+ lr_scheduler_num_cycles = 4
55
+
56
+ network_module = "networks.lora"
57
+ network_dim = 128
58
+ network_alpha = "calc:float(network_dim) / 2.0"
59
+
60
+ __upper_mixed_precision = "calc:mixed_precision.upper()"
61
+ output_name = "calc:f'Akashi_{optimizer_type}1024pxWITrue{__upper_mixed_precision}'"
62
+ log_prefix= "ref:output_name"