sdasdadas commited on
Commit
5015d77
·
1 Parent(s): ceb8f60

feat: upload zwxjirouOCV3 lora model

Browse files
zwxjirouOCV3_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 = 32
12
+ network_alpha = 1
13
+ network_args = []
14
+ network_train_unet_only = false
15
+ network_train_text_encoder_only = false
16
+
17
+ [optimizer_arguments]
18
+ min_snr_gamma = 5.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 = "zwxjirouOCV3"
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 = 8
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 = "zwxjirouOCV3"
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"
zwxjirouOCV3_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 = true
6
+ [[datasets.subsets]]
7
+ image_dir = "/content/LoRA/train_data"
8
+ num_repeats = 35
9
+
10
+
11
+ [general]
12
+ enable_bucket = true
13
+ caption_extension = ".txt"
14
+ shuffle_caption = true
15
+ keep_tokens = 1
zwxjirouOCV3_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, zwxjirou, looking at viewer, sitting, jacket, male focus, outdoors, pants, black pants"
9
+
10
+ [[prompt.subset]]
11
+ prompt = "masterpiece, best quality, zwxjirou, looking at viewer, open mouth, navel, nipples, male focus, lying, food, barefoot, spread legs, on back, armpits, feet, toes, fruit, soles, abs, pectorals, tan, towel, tanlines, bulge, foot focus, watermelon, male swimwear, swim briefs"
12
+