feat: upload lora_C3lier_MomocoV1 lora model
Browse files
lora_C3lier_MomocoV1_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/animefull-final-pruned.ckpt"
|
| 5 |
+
vae = "/content/vae/kl-f8-anime.ckpt"
|
| 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 = 16
|
| 13 |
+
network_alpha = 8
|
| 14 |
+
network_args = [ "conv_dim=8", "conv_alpha=1",]
|
| 15 |
+
network_train_unet_only = false
|
| 16 |
+
network_train_text_encoder_only = false
|
| 17 |
+
|
| 18 |
+
[optimizer_arguments]
|
| 19 |
+
min_snr_gamma = 1.0
|
| 20 |
+
optimizer_type = "AdamW8bit"
|
| 21 |
+
learning_rate = 0.0001
|
| 22 |
+
max_grad_norm = 1.0
|
| 23 |
+
optimizer_args = []
|
| 24 |
+
lr_scheduler = "constant"
|
| 25 |
+
|
| 26 |
+
[dataset_arguments]
|
| 27 |
+
cache_latents = true
|
| 28 |
+
cache_latents_to_disk = true
|
| 29 |
+
debug_dataset = false
|
| 30 |
+
vae_batch_size = 4
|
| 31 |
+
|
| 32 |
+
[training_arguments]
|
| 33 |
+
output_dir = "/content/LoRA/output"
|
| 34 |
+
output_name = "lora_C3lier_MomocoV1"
|
| 35 |
+
save_precision = "fp16"
|
| 36 |
+
save_every_n_epochs = 1
|
| 37 |
+
train_batch_size = 2
|
| 38 |
+
max_token_length = 225
|
| 39 |
+
mem_eff_attn = false
|
| 40 |
+
xformers = true
|
| 41 |
+
max_train_epochs = 8
|
| 42 |
+
max_data_loader_n_workers = 8
|
| 43 |
+
persistent_data_loader_workers = true
|
| 44 |
+
gradient_accumulation_steps = 1
|
| 45 |
+
mixed_precision = "fp16"
|
| 46 |
+
clip_skip = 2
|
| 47 |
+
ss_multires_noise_iterations = 6
|
| 48 |
+
ss_multires_noise_discount = 0.3
|
| 49 |
+
lowram = true
|
| 50 |
+
|
| 51 |
+
[logging_arguments]
|
| 52 |
+
log_with = "tensorboard"
|
| 53 |
+
logging_dir = "/content/LoRA/logs"
|
| 54 |
+
log_prefix = "lora_C3lier_MomocoV1"
|
| 55 |
+
|
| 56 |
+
[sample_prompt_arguments]
|
| 57 |
+
sample_every_n_epochs = 1
|
| 58 |
+
sample_sampler = "k_dpm_2_a"
|
| 59 |
+
|
| 60 |
+
[dreambooth_arguments]
|
| 61 |
+
prior_loss_weight = 1.0
|
| 62 |
+
|
| 63 |
+
[saving_arguments]
|
| 64 |
+
save_model_as = "safetensors"
|
lora_C3lier_MomocoV1_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 = 10
|
| 9 |
+
|
| 10 |
+
|
| 11 |
+
[general]
|
| 12 |
+
enable_bucket = true
|
| 13 |
+
caption_extension = ".txt"
|
| 14 |
+
shuffle_caption = true
|
| 15 |
+
keep_tokens = 0
|
lora_C3lier_MomocoV1_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, 1girl, solo, long hair, breasts, looking at viewer, blush, open mouth, thighhighs, hat, dress, holding, jewelry, medium breasts, green eyes, blue hair, ahoge, short sleeves, earrings, lying, frills, parted lips, puffy sleeves, black thighhighs, on back, nail polish, white dress, :o, black dress, arm up, bracelet, puffy short sleeves, zettai ryouiki, book, black headwear, no shoes, short dress, knees up"
|
| 9 |
+
|
| 10 |
+
[[prompt.subset]]
|
| 11 |
+
prompt = "masterpiece, best quality, 1girl, solo, long hair, looking at viewer, blush, open mouth, bangs, blue eyes, skirt, simple background, brown hair, shirt, hair ornament, gloves, white background, ribbon, holding, animal ears, very long hair, school uniform, collarbone, hair ribbon, white shirt, ponytail, weapon, ahoge, short sleeves, cowboy shot, pleated skirt, serafuku, choker, black gloves, hairclip, miniskirt, fingerless gloves, sailor collar, holding weapon, collar, gun, thigh strap, holding gun, rifle, o-ring, purple skirt, purple ribbon, assault rifle, m4 carbine, ar-15, purple neckerchief"
|
| 12 |
+
|