truongc63 commited on
Commit
bd59847
·
1 Parent(s): a0ae74c

feat: upload jessica_babyfat_v3 lora model

Browse files
jessica_babyfat_v3_config/config_file.toml ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Stable-Diffusion-v1-5.safetensors"
5
+ vae = "/content/vae/stablediffusion.vae.pt"
6
+
7
+ [additional_network_arguments]
8
+ no_metadata = false
9
+ unet_lr = 0.0001
10
+ text_encoder_lr = 5e-5
11
+ network_module = "networks.lora"
12
+ network_dim = 128
13
+ network_alpha = 128
14
+ network_train_unet_only = false
15
+ network_train_text_encoder_only = false
16
+
17
+ [optimizer_arguments]
18
+ optimizer_type = "AdamW8bit"
19
+ learning_rate = 0.0001
20
+ max_grad_norm = 1.0
21
+ lr_scheduler = "constant"
22
+ lr_warmup_steps = 0
23
+
24
+ [dataset_arguments]
25
+ cache_latents = true
26
+ debug_dataset = false
27
+
28
+ [training_arguments]
29
+ output_dir = "/content/LoRA/output"
30
+ output_name = "jessica_babyfat_v3"
31
+ save_precision = "fp16"
32
+ save_n_epoch_ratio = 5
33
+ train_batch_size = 3
34
+ max_token_length = 225
35
+ mem_eff_attn = false
36
+ xformers = true
37
+ max_train_epochs = 30
38
+ max_data_loader_n_workers = 8
39
+ persistent_data_loader_workers = true
40
+ gradient_checkpointing = false
41
+ gradient_accumulation_steps = 1
42
+ mixed_precision = "fp16"
43
+ clip_skip = 2
44
+ logging_dir = "/content/LoRA/logs"
45
+ log_prefix = "jessica_babyfat_v3"
46
+ noise_offset = 0.1
47
+ lowram = false
48
+
49
+ [sample_prompt_arguments]
50
+ sample_every_n_epochs = 5
51
+ sample_sampler = "k_dpm_2_a"
52
+
53
+ [dreambooth_arguments]
54
+ prior_loss_weight = 1.0
55
+
56
+ [saving_arguments]
57
+ save_model_as = "safetensors"
jessica_babyfat_v3_config/dataset_config.toml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 640
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/LoRA/train_data"
12
+ class_tokens = "jesbbfat 1girl, taiwan girl, fitness girl, a sexy woman"
13
+ num_repeats = 10
14
+
15
+ [[datasets.subsets]]
16
+ is_reg = true
17
+ image_dir = "/content/LoRA/reg_data"
18
+ class_tokens = "1girl, taiwan girl, fitness girl, a sexy woman"
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
jessica_babyfat_v3_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ jesbbfat, 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