ProjectNekoFi commited on
Commit
2b674bd
·
1 Parent(s): 5b323f2
v0-0-5_epochs_6_config/config_file.toml ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/AnyLoRA.safetensors"
5
+
6
+ [optimizer_arguments]
7
+ optimizer_type = "AdamW8bit"
8
+ learning_rate = 2e-6
9
+ max_grad_norm = 1.0
10
+ train_text_encoder = false
11
+ lr_scheduler = "constant"
12
+ lr_warmup_steps = 0
13
+
14
+ [dataset_arguments]
15
+ debug_dataset = false
16
+ in_json = "/content/fine_tune/meta_lat.json"
17
+ train_data_dir = "/content/fine_tune/train_data"
18
+ dataset_repeats = 1
19
+ shuffle_caption = true
20
+ keep_tokens = 0
21
+ resolution = "512,512"
22
+ caption_dropout_rate = 0
23
+ caption_tag_dropout_rate = 0
24
+ caption_dropout_every_n_epochs = 0
25
+ color_aug = false
26
+ token_warmup_min = 1
27
+ token_warmup_step = 0
28
+
29
+ [training_arguments]
30
+ output_dir = "/content/fine_tune/output"
31
+ output_name = "echo"
32
+ save_precision = "fp16"
33
+ save_n_epoch_ratio = 1
34
+ save_state = true
35
+ resume = "/content/eche-state"
36
+ train_batch_size = 1
37
+ max_token_length = 225
38
+ mem_eff_attn = false
39
+ xformers = true
40
+ max_train_steps = 15000
41
+ max_data_loader_n_workers = 8
42
+ persistent_data_loader_workers = true
43
+ gradient_checkpointing = false
44
+ gradient_accumulation_steps = 1
45
+ mixed_precision = "fp16"
46
+ clip_skip = 2
47
+ logging_dir = "/content/fine_tune/logs"
48
+ log_prefix = "echo"
49
+
50
+ [sample_prompt_arguments]
51
+ sample_every_n_steps = 100
52
+ sample_sampler = "ddim"
53
+
54
+ [saving_arguments]
55
+ save_model_as = "ckpt"
v0-0-5_epochs_6_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ masterpiece, best quality, 1girl, mechanical arm, science fiction, dynamic pose, white armor, black hair, chainsaw --n lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry --w 512 --h 768 --l 7 --s 28