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
|