Duskfallcrew commited on
Commit
d86f9a7
1 Parent(s): 5047792

feat: upload AkiFrontline lora model

Browse files
AkiFrontline_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/Anything-v3-1.safetensors"
5
+ vae = "/content/vae/waifudiffusion.vae.pt"
6
+
7
+ [additional_network_arguments]
8
+ no_metadata = false
9
+ unet_lr = 0.0001
10
+ text_encoder_lr = 5e-5
11
+ network_module = "lycoris.kohya"
12
+ network_dim = 32
13
+ network_alpha = 16
14
+ network_args = [ "algo=lora",]
15
+ network_train_unet_only = false
16
+ network_train_text_encoder_only = false
17
+
18
+ [optimizer_arguments]
19
+ optimizer_type = "AdamW8bit"
20
+ learning_rate = 0.0001
21
+ max_grad_norm = 1.0
22
+ lr_scheduler = "constant"
23
+ lr_warmup_steps = 0
24
+
25
+ [dataset_arguments]
26
+ cache_latents = true
27
+ debug_dataset = false
28
+
29
+ [training_arguments]
30
+ output_dir = "/content/drive/MyDrive/LoRA/output"
31
+ output_name = "AkiFrontline"
32
+ save_precision = "fp16"
33
+ save_every_n_epochs = 5
34
+ train_batch_size = 10
35
+ max_token_length = 225
36
+ mem_eff_attn = false
37
+ xformers = true
38
+ max_train_epochs = 20
39
+ max_data_loader_n_workers = 8
40
+ persistent_data_loader_workers = true
41
+ gradient_checkpointing = true
42
+ gradient_accumulation_steps = 1
43
+ mixed_precision = "fp16"
44
+ clip_skip = 2
45
+ logging_dir = "/content/LoRA/logs"
46
+ log_prefix = "AkiFrontline"
47
+ noise_offset = 0.1
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"
AkiFrontline_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 = true
9
+ color_aug = false
10
+ [[datasets.subsets]]
11
+ image_dir = "/content/LoRA/train_data"
12
+ class_tokens = "dsky_aki style"
13
+ num_repeats = 20
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 = ".txt"
25
+ shuffle_caption = true
26
+ keep_tokens = 0
27
+ bucket_reso_steps = 64
28
+ bucket_no_upscale = false
AkiFrontline_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ masterpiece, best quality, 1boy, male focus, weapon, solo, sword, holding weapon, holding, holding sword, gloves, smile, brown hair, japanese clothes, scar, short hair, black gloves, white background, jewelry, pectorals, looking at viewer, torn clothes, katana, shirt, open clothes, bangs, simple background, collarbone, grey eyes --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