sdasdadas commited on
Commit
7711cfa
·
1 Parent(s): bb03ac2

feat: upload OCv3 lora model

Browse files
OCv3_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 = 135
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 = "OCv3"
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 = 20
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 = "OCv3"
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"
OCv3_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 = 5
9
+
10
+
11
+ [general]
12
+ enable_bucket = true
13
+ caption_extension = ".txt"
14
+ shuffle_caption = true
15
+ keep_tokens = 0
OCv3_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, skirt, large breasts, simple background, shirt, long sleeves, white background, bow, holding, sitting, school uniform, white shirt, one eye closed, shoes, collared shirt, bowtie, black skirt, black footwear, red bow, book, chair, red bowtie, holding book"
9
+
10
+ [[prompt.subset]]
11
+ prompt = "masterpiece, best quality, 1girl, solo, looking at viewer, blush, simple background, white background, dress, bow, hair between eyes, brown eyes, standing, full body, short sleeves, boots, parted lips, one eye closed, shoes, puffy sleeves, sailor collar, white dress, puffy short sleeves, black bow, brown footwear, sailor dress, black sailor collar, pleated dress"
12
+