sdasdadas commited on
Commit
2760dbd
·
1 Parent(s): 13f7b32

feat: upload Lora_KronimafiaV2 lora model

Browse files
Lora_KronimafiaV2_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_bakedVae_fp16_NOTpruned.safetensors"
5
+ vae = "/content/vae/animevae.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 = 32
13
+ network_alpha = 16
14
+ network_args = []
15
+ network_train_unet_only = false
16
+ network_train_text_encoder_only = false
17
+
18
+ [optimizer_arguments]
19
+ min_snr_gamma = 5.0
20
+ optimizer_type = "AdamW8bit"
21
+ learning_rate = 0.0001
22
+ max_grad_norm = 1.0
23
+ optimizer_args = []
24
+ lr_scheduler = "constant"
25
+ lr_warmup_steps = 0
26
+
27
+ [dataset_arguments]
28
+ cache_latents = true
29
+ cache_latents_to_disk = true
30
+ debug_dataset = false
31
+ vae_batch_size = 4
32
+
33
+ [training_arguments]
34
+ output_dir = "/content/LoRA/output"
35
+ output_name = "Lora_KronimafiaV2"
36
+ save_precision = "fp16"
37
+ save_every_n_epochs = 1
38
+ train_batch_size = 2
39
+ max_token_length = 225
40
+ mem_eff_attn = false
41
+ xformers = true
42
+ max_train_epochs = 20
43
+ max_data_loader_n_workers = 8
44
+ persistent_data_loader_workers = true
45
+ gradient_accumulation_steps = 1
46
+ mixed_precision = "fp16"
47
+ clip_skip = 2
48
+ lowram = true
49
+
50
+ [logging_arguments]
51
+ log_with = "wandb"
52
+ log_tracker_name = "Lora_KronimafiaV2"
53
+ logging_dir = "/content/LoRA/logs"
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"
Lora_KronimafiaV2_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 = 20
9
+
10
+
11
+ [general]
12
+ enable_bucket = true
13
+ caption_extension = ".txt"
14
+ shuffle_caption = true
15
+ keep_tokens = 1
Lora_KronimafiaV2_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, kroniimafia, 1girl, solo, looking at viewer, blue eyes, blue hair, virtual youtuber, blood, hair intakes, black background, blood on face, black suit"
9
+
10
+ [[prompt.subset]]
11
+ prompt = "masterpiece, best quality, kroniimafia, 1girl, solo, blue eyes, sitting, blue hair, virtual youtuber, blood, blood on face, blood on clothes"
12
+