sdasdadas commited on
Commit
051a357
·
1 Parent(s): 74458ad

feat: upload winteraespa_DB_v2 checkpoint

Browse files
aespadb_akhir_config/config_file.toml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/ChilloutMix-ni-fp16.safetensors"
5
+
6
+ [optimizer_arguments]
7
+ min_snr_gamma = 5.0
8
+ optimizer_type = "AdamW8bit"
9
+ learning_rate = 1e-6
10
+ max_grad_norm = 1.0
11
+ lr_scheduler = "constant"
12
+ lr_warmup_steps = 0
13
+
14
+ [dataset_arguments]
15
+ cache_latents = true
16
+ debug_dataset = false
17
+ vae_batch_size = 1
18
+
19
+ [training_arguments]
20
+ output_dir = "/content/dreambooth/output"
21
+ output_name = "winteraespa_DB_v2"
22
+ save_precision = "fp16"
23
+ save_n_epoch_ratio = 1
24
+ save_state = false
25
+ train_batch_size = 4
26
+ max_token_length = 225
27
+ mem_eff_attn = false
28
+ xformers = true
29
+ max_train_steps = 3000
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
+ clip_skip = 1
36
+ logging_dir = "/content/dreambooth/logs"
37
+ log_prefix = "winteraespa_DB_v2"
38
+ noise_offset = 0.02
39
+
40
+ [sample_prompt_arguments]
41
+ sample_every_n_steps = 100
42
+ sample_sampler = "ddim"
43
+
44
+ [dreambooth_arguments]
45
+ prior_loss_weight = 1.0
46
+
47
+ [saving_arguments]
48
+ save_model_as = "safetensors"
aespadb_akhir_config/dataset_config.toml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 = "sls woman"
13
+ num_repeats = 50
14
+
15
+ [[datasets.subsets]]
16
+ is_reg = true
17
+ image_dir = "/content/dreambooth/reg_data"
18
+ class_tokens = "sls woman"
19
+ num_repeats = 1
20
+
21
+
22
+ [general]
23
+ enable_bucket = true
24
+ caption_extension = ".caption"
25
+ shuffle_caption = true
26
+ keep_tokens = 0
27
+ bucket_reso_steps = 64
28
+ bucket_no_upscale = false
aespadb_akhir_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