ProjectNekoFi commited on
Commit
5c92a86
1 Parent(s): 246117b

feat: upload reserval-v4 checkpoint

Browse files
reserval-v4-base_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/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-v4-base_model"
22
+ save_precision = "fp16"
23
+ save_n_epoch_ratio = 1
24
+ save_state = true
25
+ resume = "/content/dreambooth/reserval-v3-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-v4-base_model"
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"
reserval-v4-base_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
reserval-v4-base_model_config/sample_prompt.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ exceptional, best aesthetic, new, newest, best quality, masterpiece, extremely detailed, waifu, 1girl, solo --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