Commit
·
a1b83f2
1
Parent(s):
f673073
feat: upload reaimer-v0 checkpoint
Browse files- v0_config/config_file.toml +55 -0
- v0_config/meta_lat.json +0 -0
- v0_config/sample_prompt.txt +1 -0
v0_config/config_file.toml
ADDED
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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"
|
33 |
+
save_precision = "fp16"
|
34 |
+
save_n_epoch_ratio = 1
|
35 |
+
save_state = true
|
36 |
+
train_batch_size = 1
|
37 |
+
max_token_length = 225
|
38 |
+
mem_eff_attn = false
|
39 |
+
xformers = true
|
40 |
+
max_train_steps = 4500
|
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 |
+
logging_dir = "/content/fine_tune/logs"
|
47 |
+
log_prefix = "reaimer-v0"
|
48 |
+
noise_offset = 0.12
|
49 |
+
|
50 |
+
[sample_prompt_arguments]
|
51 |
+
sample_every_n_steps = 100
|
52 |
+
sample_sampler = "ddim"
|
53 |
+
|
54 |
+
[saving_arguments]
|
55 |
+
save_model_as = "safetensors"
|
v0_config/meta_lat.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
v0_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
|