ProjectNekoFi commited on
Commit
639dcde
1 Parent(s): 0f4bc4c

feat: upload reserval-v1 checkpoint

Browse files
v1_config/config_file.toml ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 = 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-v1"
22
+ save_precision = "fp16"
23
+ save_n_epoch_ratio = 1
24
+ save_state = true
25
+ train_batch_size = 2
26
+ max_token_length = 225
27
+ mem_eff_attn = false
28
+ xformers = true
29
+ max_train_steps = 5000
30
+ max_data_loader_n_workers = 8
31
+ persistent_data_loader_workers = true
32
+ gradient_checkpointing = false
33
+ gradient_accumulation_steps = 1
34
+ mixed_precision = "fp16"
35
+ logging_dir = "/content/dreambooth/logs"
36
+ log_prefix = "reserval-v1"
37
+
38
+ [sample_prompt_arguments]
39
+ sample_every_n_steps = 100
40
+ sample_sampler = "ddim"
41
+
42
+ [dreambooth_arguments]
43
+ prior_loss_weight = 1.0
44
+
45
+ [saving_arguments]
46
+ save_model_as = "safetensors"
v1_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
v1_config/sample_prompt.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ exceptional, best aesthetic, new, newest, best quality, masterpiece, extremely detailed, waifu, 1girl, bare shoulders, black dress, black pantyhose, blue dress, bow choker, cone hair bun, detached sleeves, diamond-shaped pupils, diamond (shape), dress, genshin impact, hair bun, hair intakes, hair ornament, highres, keqing (genshin impact), keqing (opulent splendor) (genshin impact), long hair, official alternate costume, pantyhose, purple eyes, purple hair, solo, standing, strapless, strapless dress, symbol-shaped pupils, twintails, two-tone dress, wrist cuffs, --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