feat: upload k_m_swimsuit_v4 lora model
Browse files
k_m_swimsuit_v4_config/config_file.toml
ADDED
|
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[sdxl_arguments]
|
| 2 |
+
cache_text_encoder_outputs = false
|
| 3 |
+
no_half_vae = true
|
| 4 |
+
min_timestep = 0
|
| 5 |
+
max_timestep = 1000
|
| 6 |
+
shuffle_caption = true
|
| 7 |
+
lowram = true
|
| 8 |
+
|
| 9 |
+
[model_arguments]
|
| 10 |
+
pretrained_model_name_or_path = "hanungaddi/zipsydiffv2"
|
| 11 |
+
vae = "/content/vae/sdxl_vae.safetensors"
|
| 12 |
+
|
| 13 |
+
[dataset_arguments]
|
| 14 |
+
debug_dataset = false
|
| 15 |
+
in_json = "/content/LoRA/meta_lat.json"
|
| 16 |
+
train_data_dir = "/content/LoRA/train_data"
|
| 17 |
+
dataset_repeats = 1
|
| 18 |
+
keep_tokens = 0
|
| 19 |
+
resolution = "1024,1024"
|
| 20 |
+
color_aug = false
|
| 21 |
+
token_warmup_min = 1
|
| 22 |
+
token_warmup_step = 0
|
| 23 |
+
|
| 24 |
+
[training_arguments]
|
| 25 |
+
output_dir = "/content/LoRA/output/k_m_swimsuit_v4"
|
| 26 |
+
output_name = "k_m_swimsuit_v4"
|
| 27 |
+
save_precision = "fp16"
|
| 28 |
+
save_every_n_epochs = 2
|
| 29 |
+
train_batch_size = 2
|
| 30 |
+
max_token_length = 225
|
| 31 |
+
mem_eff_attn = false
|
| 32 |
+
sdpa = false
|
| 33 |
+
xformers = true
|
| 34 |
+
max_train_epochs = 20
|
| 35 |
+
max_data_loader_n_workers = 8
|
| 36 |
+
persistent_data_loader_workers = true
|
| 37 |
+
gradient_checkpointing = true
|
| 38 |
+
gradient_accumulation_steps = 1
|
| 39 |
+
mixed_precision = "fp16"
|
| 40 |
+
cache_latents = true
|
| 41 |
+
cache_latents_to_disk = true
|
| 42 |
+
|
| 43 |
+
[logging_arguments]
|
| 44 |
+
log_with = "wandb"
|
| 45 |
+
log_tracker_name = "k_m_swimsuit_v4"
|
| 46 |
+
logging_dir = "/content/LoRA/logs"
|
| 47 |
+
|
| 48 |
+
[sample_prompt_arguments]
|
| 49 |
+
sample_every_n_epochs = 2
|
| 50 |
+
sample_sampler = "euler_a"
|
| 51 |
+
|
| 52 |
+
[saving_arguments]
|
| 53 |
+
save_model_as = "safetensors"
|
| 54 |
+
|
| 55 |
+
[optimizer_arguments]
|
| 56 |
+
optimizer_type = "AdaFactor"
|
| 57 |
+
learning_rate = 0.0003
|
| 58 |
+
network_train_unet_only = true
|
| 59 |
+
max_grad_norm = 0
|
| 60 |
+
optimizer_args = [ "scale_parameter=False", "relative_step=False", "warmup_init=False",]
|
| 61 |
+
lr_scheduler = "cosine_with_restarts"
|
| 62 |
+
lr_warmup_steps = 100
|
| 63 |
+
lr_scheduler_num_cycles = 2
|
| 64 |
+
|
| 65 |
+
[additional_network_arguments]
|
| 66 |
+
no_metadata = false
|
| 67 |
+
network_module = "networks.lora"
|
| 68 |
+
network_dim = 32
|
| 69 |
+
network_alpha = 16
|
| 70 |
+
network_args = [ "conv_dim=16", "conv_alpha=8",]
|
| 71 |
+
|
| 72 |
+
[advanced_training_config]
|
| 73 |
+
save_state = false
|
| 74 |
+
save_last_n_epochs_state = false
|
| 75 |
+
caption_dropout_rate = 0
|
| 76 |
+
caption_tag_dropout_rate = 0.5
|
| 77 |
+
caption_dropout_every_n_epochs = 0
|
| 78 |
+
min_snr_gamma = 8
|
k_m_swimsuit_v4_config/sample_prompt.toml
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[prompt]
|
| 2 |
+
negative_prompt = "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, "
|
| 3 |
+
width = 1024
|
| 4 |
+
height = 1024
|
| 5 |
+
scale = 12
|
| 6 |
+
sample_steps = 28
|
| 7 |
+
[[prompt.subset]]
|
| 8 |
+
prompt = "hiten, masterpiece, best quality, 1girl, honkai (series), honkai impact 3rd, raiden mei, raiden mei (herrscher of origin), raiden mei (crooning tides), 1girl, arm ribbon, blue one-piece swimsuit, bow, braid, breasts, casual one-piece swimsuit, closed mouth, cowboy shot, earrings, flower, hair bun, hair flower, hair ornament, hairband, jewelry, large breasts, one-piece swimsuit, one eye closed, purple ribbon, ribbon, rose, simple background, smile, solo, swimsuit, thigh strap, white background, white bow, white flower, white hairband, white rose, wrist bow, wrist ribbon"
|
| 9 |
+
|
| 10 |
+
[[prompt.subset]]
|
| 11 |
+
prompt = "hiten, masterpiece, best quality, kiana kaslana (selenic ripples), 1girl, animal ears, bare shoulders, bikini, black bikini, black bow, black sarong, blue eyes, bow, breasts, cat ears, cleavage, criss-cross straps, hair between eyes, hair bow, hand up, large breasts, looking at viewer, navel, official alternate costume, purple pupils, sarong, star-shaped pupils, star (symbol), stomach, swimsuit, see-through clothes, swimsuit cover-up, symbol-shaped pupils, white hair, white wrist cuffs, wrist cuffs"
|
| 12 |
+
|
| 13 |
+
[[prompt.subset]]
|
| 14 |
+
prompt = "hiten, masterpiece, best quality, honkai (series), honkai impact 3rd, kiana kaslana, kiana kaslana (herrscher of the void), kiana kaslana (parasol kaiserin), 1girl, ahoge, alternate costume, aqua dress, bare shoulders, beach, breasts, detached sleeves, diamond-shaped pupils, diamond (shape), dress, earrings, hair between eyes, halter dress, halterneck, innertube, jewelry, large breasts, long hair, looking at viewer, parted lips, pleated dress, sidelocks, solo, starfish, swim ring, symbol-shaped pupils, two-tone dress, white dress, white hair, wristband, yellow eyes, white background, simple background"
|
| 15 |
+
|
| 16 |
+
|