ProjectNekoFi commited on
Commit
522274b
1 Parent(s): 09d5cba

feat: upload echo checkpoint

Browse files
v0-0-1_epochs_0-8_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 = true
35
+ train_batch_size = 1
36
+ max_token_length = 225
37
+ mem_eff_attn = false
38
+ xformers = true
39
+ max_train_steps = 5000
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"
v0-0-1_epochs_0-8_config/meta_clean.json ADDED
The diff for this file is too large to render. See raw diff
 
v0-0-1_epochs_0-8_config/meta_lat.json ADDED
The diff for this file is too large to render. See raw diff
 
v0-0-1_epochs_0-8_config/sample_prompt.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ highres, 1girl, science fiction, mechanical --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