feat: upload matsuriv2 checkpoint
Browse files
Matsuriv2-dataset-true_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-V2.0_fp16.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 |
+
train_text_encoder = false
|
12 |
+
lr_scheduler = "constant"
|
13 |
+
lr_warmup_steps = 0
|
14 |
+
|
15 |
+
[dataset_arguments]
|
16 |
+
enable_bucket = true
|
17 |
+
debug_dataset = false
|
18 |
+
in_json = "/content/fine_tune/config/meta_lat.json"
|
19 |
+
train_data_dir = "/content/fine_tune/train_data"
|
20 |
+
dataset_repeats = 1
|
21 |
+
shuffle_caption = true
|
22 |
+
keep_tokens = 0
|
23 |
+
resolution = "768,768"
|
24 |
+
caption_dropout_rate = 0
|
25 |
+
caption_tag_dropout_rate = 0
|
26 |
+
caption_dropout_every_n_epochs = 0
|
27 |
+
color_aug = false
|
28 |
+
token_warmup_min = 1
|
29 |
+
token_warmup_step = 0
|
30 |
+
|
31 |
+
[training_arguments]
|
32 |
+
output_dir = "/content/fine_tune/output"
|
33 |
+
output_name = "matsuriv2"
|
34 |
+
save_precision = "fp16"
|
35 |
+
save_n_epoch_ratio = 10
|
36 |
+
save_state = false
|
37 |
+
train_batch_size = 1
|
38 |
+
max_token_length = 225
|
39 |
+
mem_eff_attn = false
|
40 |
+
xformers = true
|
41 |
+
max_train_steps = 5000
|
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 = "matsuriv2"
|
49 |
+
|
50 |
+
[sample_prompt_arguments]
|
51 |
+
sample_every_n_steps = 250
|
52 |
+
sample_sampler = "ddim"
|
53 |
+
|
54 |
+
[saving_arguments]
|
55 |
+
save_model_as = "ckpt"
|
Matsuriv2-dataset-true_config/meta_lat.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
Matsuriv2-dataset-true_config/sample_prompt.txt
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
|
2 |
+
masterpiece, best quality, 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
|