feat: upload Locon_SuteinuA_V4 lora model
Browse files
Locon_SuteinuA_V4_config/config_file.toml
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[model_arguments]
|
| 2 |
+
v2 = false
|
| 3 |
+
v_parameterization = false
|
| 4 |
+
pretrained_model_name_or_path = "/content/pretrained_model/anyloraCheckpoint_novaeFp16.safetensors"
|
| 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 = 16
|
| 12 |
+
network_alpha = 8
|
| 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 = "cosine"
|
| 24 |
+
lr_warmup_steps = 0
|
| 25 |
+
|
| 26 |
+
[dataset_arguments]
|
| 27 |
+
cache_latents_to_disk = true
|
| 28 |
+
debug_dataset = false
|
| 29 |
+
vae_batch_size = 4
|
| 30 |
+
|
| 31 |
+
[training_arguments]
|
| 32 |
+
output_dir = "/content/LoRA/output"
|
| 33 |
+
output_name = "Locon_SuteinuA_V4"
|
| 34 |
+
save_precision = "fp16"
|
| 35 |
+
save_every_n_epochs = 1
|
| 36 |
+
train_batch_size = 2
|
| 37 |
+
max_token_length = 225
|
| 38 |
+
mem_eff_attn = false
|
| 39 |
+
xformers = true
|
| 40 |
+
max_train_epochs = 3
|
| 41 |
+
max_data_loader_n_workers = 8
|
| 42 |
+
persistent_data_loader_workers = true
|
| 43 |
+
seed = 544655
|
| 44 |
+
gradient_checkpointing = false
|
| 45 |
+
gradient_accumulation_steps = 1
|
| 46 |
+
mixed_precision = "fp16"
|
| 47 |
+
clip_skip = 2
|
| 48 |
+
noise_offset = 0.2
|
| 49 |
+
lowram = true
|
| 50 |
+
|
| 51 |
+
[logging_arguments]
|
| 52 |
+
log_with = "tensorboard"
|
| 53 |
+
logging_dir = "/content/LoRA/logs"
|
| 54 |
+
log_prefix = "Locon_SuteinuA_V4"
|
| 55 |
+
|
| 56 |
+
[dreambooth_arguments]
|
| 57 |
+
prior_loss_weight = 1.0
|
| 58 |
+
|
| 59 |
+
[saving_arguments]
|
| 60 |
+
save_model_as = "safetensors"
|
| 61 |
+
|
| 62 |
+
[sample_prompt_arguments]
|
| 63 |
+
sample_every_n_epochs = 1
|
| 64 |
+
sample_sampler = "ddim"
|
Locon_SuteinuA_V4_config/dataset_config.toml
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 = false
|
| 9 |
+
color_aug = false
|
| 10 |
+
[[datasets.subsets]]
|
| 11 |
+
image_dir = "/content/LoRA/train_data"
|
| 12 |
+
class_tokens = "SuteinuA"
|
| 13 |
+
num_repeats = 25
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
[general]
|
| 17 |
+
enable_bucket = true
|
| 18 |
+
caption_extension = ".txt"
|
| 19 |
+
shuffle_caption = true
|
| 20 |
+
keep_tokens = 1
|
| 21 |
+
bucket_reso_steps = 64
|
| 22 |
+
bucket_no_upscale = false
|
Locon_SuteinuA_V4_config/sample_prompt.txt
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
masterpiece, best quality, SuteinuA, 1girl, solo, long hair, looking at viewer, smile, bangs, 1boy, hair between eyes, closed mouth, blue hair, yellow eyes, braid, male focus, blurry, portrait --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|
| 2 |
+
masterpiece, best quality, SuteinuA, 1girl, solo, looking at viewer, smile, bangs, brown hair, shirt, hat, holding, twintails, brown eyes, jewelry, closed mouth, collarbone, white shirt, earrings, portrait, spoon, grey headwear, holding spoon, plaid headwear --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|
| 3 |
+
masterpiece, best quality, SuteinuA, 1girl, solo, smile, short hair, open mouth, blonde hair, shirt, hair ornament, long sleeves, hat, bow, yellow eyes, upper body, hairclip, bowtie, orange eyes, white headwear, eyepatch, yellow bow --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|
| 4 |
+
masterpiece, best quality, SuteinuA, 1girl, solo, long hair, looking at viewer, blush, smile, bangs, red eyes, hat, ribbon, twintails, closed mouth, hair ribbon, upper body, pink hair, frills, food, japanese clothes, blunt bangs, pink eyes, kimono, apron, blood, :3, fruit, red background, blood on face, strawberry, triangular headpiece, blood on clothes, saigyouji yuyuko --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|
| 5 |
+
masterpiece, best quality, SuteinuA, 1girl, solo, long hair, looking at viewer, open mouth, bangs, brown hair, hair ornament, bow, purple eyes, braid, flower, hair bow, parted lips, choker, virtual youtuber, hair flower, hair bun, star (symbol), mole, twin braids, mole under eye, double bun, portrait, hand on own face, hand on own cheek --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|