ProjectNekoFi commited on
Commit
548c3a9
1 Parent(s): a2a2055

feat: upload reaimer-v0-e2-5 checkpoint

Browse files
reaimer-v0-e2-5_config/config_file.toml ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = true
3
+ v_parameterization = true
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Replicant-V3.0_fp16.safetensors"
5
+
6
+ [optimizer_arguments]
7
+ min_snr_gamma = 5
8
+ optimizer_type = "AdamW8bit"
9
+ learning_rate = 1.5e-6
10
+ max_grad_norm = 1.0
11
+ train_text_encoder = true
12
+ lr_scheduler = "constant"
13
+ lr_warmup_steps = 0
14
+
15
+ [dataset_arguments]
16
+ debug_dataset = false
17
+ in_json = "/content/fine_tune/meta_lat.json"
18
+ train_data_dir = "/content/fine_tune/train_data"
19
+ dataset_repeats = 1
20
+ shuffle_caption = true
21
+ keep_tokens = 0
22
+ resolution = "768,768"
23
+ caption_dropout_rate = 0
24
+ caption_tag_dropout_rate = 0
25
+ caption_dropout_every_n_epochs = 0
26
+ color_aug = false
27
+ token_warmup_min = 1
28
+ token_warmup_step = 0
29
+
30
+ [training_arguments]
31
+ output_dir = "/content/fine_tune/output"
32
+ output_name = "reaimer-v0-e2-5"
33
+ save_precision = "fp16"
34
+ save_n_epoch_ratio = 1
35
+ save_state = true
36
+ resume = "/content/fine_tune/reaimer-state"
37
+ train_batch_size = 1
38
+ max_token_length = 225
39
+ mem_eff_attn = false
40
+ xformers = true
41
+ max_train_steps = 9000
42
+ max_data_loader_n_workers = 8
43
+ persistent_data_loader_workers = true
44
+ gradient_checkpointing = false
45
+ gradient_accumulation_steps = 1
46
+ mixed_precision = "fp16"
47
+ logging_dir = "/content/fine_tune/logs"
48
+ log_prefix = "reaimer-v0-e2-5"
49
+ noise_offset = 0.12
50
+
51
+ [sample_prompt_arguments]
52
+ sample_every_n_steps = 100
53
+ sample_sampler = "ddim"
54
+
55
+ [saving_arguments]
56
+ save_model_as = "safetensors"
reaimer-v0-e2-5_config/sample_prompt.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ exceptional, best aesthetic, new, newest, best quality, masterpiece, extremely detailed, waifu, 1girl, solo, selfie --n simple background, white background, lowres, bad anatomy, bad hands, missing finger, extra digits, fewer digits, blurry, mutated hands and fingers, poorly drawn face, mutation, deformed face, ugly, bad proportions, extra limbs, extra face, double head, extra head, extra feet, monster, logo, cropped, worst quality, jpeg, humpbacked, long body, long neck, jpeg artifacts, deleted, old, oldest, censored,bad aesthetic, mosaic censoring, bar censor, blur censor, simple background --w 512 --h 768 --l 7 --s 28