sdasdadas commited on
Commit
a072958
·
1 Parent(s): d5acf79

feat: upload OCv2 lora model

Browse files
OCv2_config/config_file.toml ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/AnyLoRA_noVae_fp16-pruned.ckpt"
5
+
6
+ [additional_network_arguments]
7
+ no_metadata = false
8
+ unet_lr = 0.0001
9
+ text_encoder_lr = 5e-5
10
+ network_module = "networks.lora"
11
+ network_dim = 64
12
+ network_alpha = 32
13
+ network_args = []
14
+ network_train_unet_only = false
15
+ network_train_text_encoder_only = false
16
+
17
+ [optimizer_arguments]
18
+ min_snr_gamma = 1.0
19
+ optimizer_type = "AdamW8bit"
20
+ learning_rate = 0.0001
21
+ max_grad_norm = 1.0
22
+ optimizer_args = []
23
+ lr_scheduler = "constant_with_warmup"
24
+ lr_warmup_steps = 35
25
+
26
+ [dataset_arguments]
27
+ cache_latents = true
28
+ cache_latents_to_disk = true
29
+ debug_dataset = false
30
+ vae_batch_size = 4
31
+
32
+ [training_arguments]
33
+ output_dir = "/content/LoRA/output"
34
+ output_name = "OCv2"
35
+ save_precision = "fp16"
36
+ save_every_n_epochs = 1
37
+ train_batch_size = 2
38
+ max_token_length = 225
39
+ mem_eff_attn = false
40
+ xformers = true
41
+ max_train_epochs = 5
42
+ max_data_loader_n_workers = 8
43
+ persistent_data_loader_workers = true
44
+ seed = 46
45
+ gradient_accumulation_steps = 1
46
+ mixed_precision = "fp16"
47
+ clip_skip = 2
48
+ lowram = true
49
+
50
+ [logging_arguments]
51
+ log_with = "tensorboard"
52
+ logging_dir = "/content/LoRA/logs"
53
+ log_prefix = "OCv2"
54
+
55
+ [sample_prompt_arguments]
56
+ sample_every_n_epochs = 1
57
+ sample_sampler = "k_dpm_2_a"
58
+
59
+ [dreambooth_arguments]
60
+ prior_loss_weight = 1.0
61
+
62
+ [saving_arguments]
63
+ save_model_as = "safetensors"
OCv2_config/dataset_config.toml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 512
3
+ min_bucket_reso = 256
4
+ max_bucket_reso = 1024
5
+ flip_aug = false
6
+ [[datasets.subsets]]
7
+ image_dir = "/content/LoRA/train_data"
8
+ num_repeats = 50
9
+
10
+
11
+ [general]
12
+ enable_bucket = true
13
+ caption_extension = ".txt"
14
+ shuffle_caption = true
15
+ keep_tokens = 0
OCv2_config/sample_prompt.toml ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [prompt]
2
+ negative_prompt = "(worst quality, low quality:1.4), "
3
+ width = 512
4
+ height = 768
5
+ scale = 7
6
+ sample_steps = 28
7
+ [[prompt.subset]]
8
+ prompt = "masterpiece, best quality, 1girl, solo, breasts, looking at viewer, blush, simple background, long sleeves, white background, navel, medium breasts, nipples, standing, collarbone, full body, boots, one eye closed, pussy, coat, sleeves past wrists, grey eyes, brown footwear, thigh gap, cleft of venus, naked coat, naked jacket"
9
+
10
+ [[prompt.subset]]
11
+ prompt = "masterpiece, best quality, 1girl, solo, breasts, looking at viewer, blush, simple background, white background, dress, holding, standing, collarbone, full body, flower, short sleeves, parted lips, one eye closed, puffy sleeves, black footwear, white dress, puffy short sleeves, grey eyes, rose, pink flower, bouquet, wedding dress, pink rose, holding bouquet"
12
+