ProjectNekoFi commited on
Commit
5bb4ccb
·
1 Parent(s): fba7754

feat: upload echo checkpoint

Browse files
v0-0-3_epochs_2-8_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/echo-diffusion/v0-0-1_epochs_0-8_config/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/echo-diffusion/v0-0-1_epochs_0-8/echo-state"
36
+ train_batch_size = 1
37
+ max_token_length = 225
38
+ mem_eff_attn = false
39
+ xformers = true
40
+ max_train_steps = 12000
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-3_epochs_2-8_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ masterpiece, best quality, science fiction, mechanical, mecha, 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