ProjectNekoFi commited on
Commit
e60bef8
·
1 Parent(s): 65ebe28

feat: upload echo checkpoint

Browse files
newdata_config/config_file.toml ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 = false
35
+ train_batch_size = 1
36
+ max_token_length = 225
37
+ mem_eff_attn = false
38
+ xformers = true
39
+ max_train_steps = 2500
40
+ max_data_loader_n_workers = 8
41
+ persistent_data_loader_workers = true
42
+ gradient_checkpointing = false
43
+ gradient_accumulation_steps = 1
44
+ mixed_precision = "fp16"
45
+ clip_skip = 2
46
+ logging_dir = "/content/fine_tune/logs"
47
+ log_prefix = "echo"
48
+
49
+ [sample_prompt_arguments]
50
+ sample_every_n_steps = 100
51
+ sample_sampler = "ddim"
52
+
53
+ [saving_arguments]
54
+ save_model_as = "ckpt"
newdata_config/meta_clean.json ADDED
The diff for this file is too large to render. See raw diff
 
newdata_config/meta_lat.json ADDED
The diff for this file is too large to render. See raw diff
 
newdata_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ masterpiece, best quality, 1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt --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