sdasdadas commited on
Commit
48051c6
·
1 Parent(s): a354dfc

feat: upload Dreambooth_pattern_5conceptV1 checkpoint

Browse files
patternV1_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/AnyLoRA.safetensors"
5
+
6
+ [optimizer_arguments]
7
+ min_snr_gamma = 5.0
8
+ optimizer_type = "AdamW8bit"
9
+ learning_rate = 2e-6
10
+ max_grad_norm = 1.0
11
+ lr_scheduler = "constant_with_warmup"
12
+ lr_warmup_steps = 5
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 = "Dreambooth_pattern_5conceptV1"
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 = 2500
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 = 2
36
+ logging_dir = "/content/dreambooth/logs"
37
+ log_prefix = "Dreambooth_pattern_5conceptV1"
38
+ noise_offset = 0.05
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"
patternV1_config/dataset_config.toml ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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/concept3_camouflage"
12
+ class_tokens = "pttrncamouflage"
13
+ num_repeats = 15
14
+
15
+ [[datasets.subsets]]
16
+ image_dir = "/content/dreambooth/train_data/concept5_paisley"
17
+ class_tokens = "pttrnpaisley"
18
+ num_repeats = 20
19
+
20
+ [[datasets.subsets]]
21
+ image_dir = "/content/dreambooth/train_data/concept1_leopard"
22
+ class_tokens = "pttrnleopard"
23
+ num_repeats = 25
24
+
25
+ [[datasets.subsets]]
26
+ image_dir = "/content/dreambooth/train_data/concept2_zebra"
27
+ class_tokens = "pttrnzebra"
28
+ num_repeats = 18
29
+
30
+ [[datasets.subsets]]
31
+ image_dir = "/content/dreambooth/train_data/concept4_floraldisty"
32
+ class_tokens = "pttrnfloraldisty"
33
+ num_repeats = 22
34
+
35
+
36
+ [general]
37
+ enable_bucket = true
38
+ caption_extension = ".caption"
39
+ shuffle_caption = true
40
+ keep_tokens = 1
41
+ bucket_reso_steps = 64
42
+ bucket_no_upscale = false
patternV1_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ pttrnleopard, 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