ProjectNekoFi
commited on
Commit
•
3297379
1
Parent(s):
ce20cc1
feat: upload reserval-v3 checkpoint
Browse files
v3-last_model_config/config_file.toml
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[model_arguments]
|
2 |
+
v2 = true
|
3 |
+
v_parameterization = true
|
4 |
+
pretrained_model_name_or_path = "/content/pretrained_model/reserval-v2.safetensors"
|
5 |
+
|
6 |
+
[optimizer_arguments]
|
7 |
+
min_snr_gamma = 5
|
8 |
+
optimizer_type = "AdamW8bit"
|
9 |
+
learning_rate = 2e-6
|
10 |
+
max_grad_norm = 1.0
|
11 |
+
lr_scheduler = "constant"
|
12 |
+
lr_warmup_steps = 0
|
13 |
+
|
14 |
+
[dataset_arguments]
|
15 |
+
cache_latens = false
|
16 |
+
debug_dataset = false
|
17 |
+
vae_batch_size = 1
|
18 |
+
|
19 |
+
[training_arguments]
|
20 |
+
output_dir = "/content/dreambooth/output"
|
21 |
+
output_name = "reserval-v3"
|
22 |
+
save_precision = "fp16"
|
23 |
+
save_n_epoch_ratio = 1
|
24 |
+
save_state = true
|
25 |
+
resume = "/content/dreambooth/reserval-v2-state"
|
26 |
+
train_batch_size = 2
|
27 |
+
max_token_length = 225
|
28 |
+
mem_eff_attn = false
|
29 |
+
xformers = true
|
30 |
+
max_train_steps = 7000
|
31 |
+
max_data_loader_n_workers = 8
|
32 |
+
persistent_data_loader_workers = true
|
33 |
+
gradient_checkpointing = false
|
34 |
+
gradient_accumulation_steps = 1
|
35 |
+
mixed_precision = "fp16"
|
36 |
+
logging_dir = "/content/dreambooth/logs"
|
37 |
+
log_prefix = "reserval-v3"
|
38 |
+
|
39 |
+
[sample_prompt_arguments]
|
40 |
+
sample_every_n_steps = 100
|
41 |
+
sample_sampler = "ddim"
|
42 |
+
|
43 |
+
[dreambooth_arguments]
|
44 |
+
prior_loss_weight = 1.0
|
45 |
+
|
46 |
+
[saving_arguments]
|
47 |
+
save_model_as = "safetensors"
|
v3-last_model_config/dataset_config.toml
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[[datasets]]
|
2 |
+
resolution = 512
|
3 |
+
min_bucket_reso = 256
|
4 |
+
max_bucket_reso = 1024
|
5 |
+
caption_dropout_rate = 0
|
6 |
+
caption_tag_dropout_rate = 0
|
7 |
+
caption_dropout_every_n_epochs = 0
|
8 |
+
flip_aug = false
|
9 |
+
color_aug = false
|
10 |
+
[[datasets.subsets]]
|
11 |
+
image_dir = "/content/dreambooth/train_data"
|
12 |
+
class_tokens = "mksks style"
|
13 |
+
num_repeats = 2
|
14 |
+
|
15 |
+
|
16 |
+
[general]
|
17 |
+
enable_bucket = true
|
18 |
+
caption_extension = ".txt"
|
19 |
+
shuffle_caption = true
|
20 |
+
keep_tokens = 0
|
21 |
+
bucket_reso_steps = 64
|
22 |
+
bucket_no_upscale = false
|
v3-last_model_config/sample_prompt.txt
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
exceptional, best aesthetic, new, newest, best quality, masterpiece, extremely detailed, waifu, 1girl, bare shoulders, black hair, ponytail, shinomiya kaguya, black school uniform, red eyes, open eyes, short hair with long sidelocks, red ribbon, hair ribbon --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
|