sdasdadas commited on
Commit
3bbb154
·
1 Parent(s): 3621182

feat: upload rachelV2 lora model

Browse files
rachelV2_config/config_file.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Chillout-mix.safetensors"
5
+
6
+ [additional_network_arguments]
7
+ no_metadata = false
8
+ unet_lr = 0.0001
9
+ text_encoder_lr = 5e-5
10
+ network_module = "lycoris.kohya"
11
+ network_dim = 16
12
+ network_alpha = 8
13
+ network_args = [ "conv_dim=8", "conv_alpha=1", "algo=lora",]
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 = "rachelV2"
31
+ save_precision = "fp16"
32
+ save_every_n_epochs = 1
33
+ train_batch_size = 6
34
+ max_token_length = 225
35
+ mem_eff_attn = false
36
+ xformers = true
37
+ max_train_epochs = 25
38
+ max_data_loader_n_workers = 8
39
+ persistent_data_loader_workers = true
40
+ seed = 1234
41
+ gradient_checkpointing = false
42
+ gradient_accumulation_steps = 1
43
+ mixed_precision = "fp16"
44
+ clip_skip = 1
45
+ logging_dir = "/content/LoRA/logs"
46
+ log_prefix = "rachelV2"
47
+ noise_offset = 0.05
48
+ lowram = true
49
+
50
+ [sample_prompt_arguments]
51
+ sample_every_n_epochs = 1
52
+ sample_sampler = "ddim"
53
+
54
+ [dreambooth_arguments]
55
+ prior_loss_weight = 1.0
56
+
57
+ [saving_arguments]
58
+ save_model_as = "safetensors"
rachelV2_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/LoRA/train_data"
12
+ class_tokens = "mksks style"
13
+ num_repeats = 15
14
+
15
+ [[datasets.subsets]]
16
+ is_reg = true
17
+ image_dir = "/content/LoRA/reg_data"
18
+ class_tokens = "style"
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
rachelV2_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