feat: upload odakaori_50repeat_3epoch_V1 lora model
Browse files
odakaori_50repeat_3epoch_V1_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/chilloutmix_NiPrunedFp16Fix.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 = [ "conv_dim=8", "conv_alpha=1",]
|
| 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 = "odakaori_50repeat_3epoch_V1"
|
| 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 = 546656
|
| 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 = "wandb"
|
| 53 |
+
log_tracker_name = "odakaori_50repeat_3epoch_V1"
|
| 54 |
+
logging_dir = "/content/LoRA/logs"
|
| 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"
|
odakaori_50repeat_3epoch_V1_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 = "odakaori"
|
| 13 |
+
num_repeats = 50
|
| 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
|
odakaori_50repeat_3epoch_V1_config/sample_prompt.txt
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
masterpiece, best quality, 1girl, solo, long hair, looking at viewer, bangs, shirt, black hair, closed mouth, white shirt, upper body, glasses, collared shirt, black eyes, lips, wing collar, red-framed eyewear, realistic --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|
| 2 |
+
masterpiece, best quality, 1girl, solo, smile, brown hair, scarf, grin, cosplay, parody, realistic, real life insert, voice actor --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|
| 3 |
+
masterpiece, best quality, 1girl, solo, long hair, looking at viewer, smile, bangs, shirt, black hair, upper body, blunt bangs, black eyes, lips, realistic --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|
| 4 |
+
masterpiece, best quality, 1girl, solo, long hair, looking at viewer, bangs, shirt, black hair, dress, bare shoulders, jewelry, white shirt, upper body, parted lips, necklace, off shoulder, blurry, black eyes, lips, realistic --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|
| 5 |
+
masterpiece, best quality, 1girl, solo, long hair, looking at viewer, simple background, brown hair, black hair, dress, jewelry, upper body, earrings, necklace, black eyes, black dress, lips, black background, realistic --n (worst quality, low quality:1.4) --w 512 --h 768 --l 7 --s 28
|