ProjectNekoFi commited on
Commit
6e4d570
·
1 Parent(s): d1c6071

feat: upload lootbox3 checkpoint

Browse files
lootbox3_config/config_file.toml ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Anything-v3-1.safetensors"
5
+
6
+ [optimizer_arguments]
7
+ optimizer_type = "AdamW8bit"
8
+ learning_rate = 2e-6
9
+ max_grad_norm = 1.0
10
+ lr_scheduler = "constant"
11
+ lr_warmup_steps = 0
12
+
13
+ [dataset_arguments]
14
+ cache_latents = true
15
+ debug_dataset = false
16
+
17
+ [training_arguments]
18
+ output_dir = "/content/dreambooth/output"
19
+ output_name = "lootbox3"
20
+ save_precision = "fp16"
21
+ save_n_epoch_ratio = 1
22
+ save_state = true
23
+ resume = "/content/dreambooth/output/lootbox2-state"
24
+ train_batch_size = 1
25
+ max_token_length = 225
26
+ mem_eff_attn = false
27
+ xformers = true
28
+ max_train_steps = 8000
29
+ max_data_loader_n_workers = 8
30
+ persistent_data_loader_workers = true
31
+ gradient_checkpointing = false
32
+ gradient_accumulation_steps = 1
33
+ mixed_precision = "fp16"
34
+ clip_skip = 2
35
+ logging_dir = "/content/dreambooth/logs"
36
+ log_prefix = "lootbox3"
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 = "ckpt"
lootbox3_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 = true
9
+ color_aug = false
10
+ [[datasets.subsets]]
11
+ image_dir = "/content/dreambooth/train_data"
12
+ class_tokens = "lootbox style"
13
+ num_repeats = 50
14
+
15
+ [[datasets.subsets]]
16
+ is_reg = true
17
+ image_dir = "/content/dreambooth/reg_data"
18
+ class_tokens = "style"
19
+ num_repeats = 1
20
+
21
+
22
+ [general]
23
+ enable_bucket = true
24
+ caption_extension = ".txt"
25
+ shuffle_caption = true
26
+ keep_tokens = 0
27
+ bucket_reso_steps = 64
28
+ bucket_no_upscale = false
lootbox3_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