feat: upload hsrSplashArtv2 lora model
Browse files
hsrSplashArtv2_config/config_file.toml
ADDED
|
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[model_arguments]
|
| 2 |
+
v2 = false
|
| 3 |
+
v_parameterization = false
|
| 4 |
+
pretrained_model_name_or_path = "/content/pretrained_model/eclipseV10Asami.safetensors"
|
| 5 |
+
vae = "/content/vae/anime.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 = "networks.lora"
|
| 12 |
+
network_dim = 64
|
| 13 |
+
network_alpha = 64
|
| 14 |
+
network_args = []
|
| 15 |
+
network_train_unet_only = false
|
| 16 |
+
network_train_text_encoder_only = false
|
| 17 |
+
|
| 18 |
+
[optimizer_arguments]
|
| 19 |
+
optimizer_type = "Lion"
|
| 20 |
+
learning_rate = 0.0001
|
| 21 |
+
max_grad_norm = 1.0
|
| 22 |
+
optimizer_args = []
|
| 23 |
+
lr_scheduler = "cosine_with_restarts"
|
| 24 |
+
lr_warmup_steps = 0
|
| 25 |
+
lr_scheduler_num_cycles = 0
|
| 26 |
+
|
| 27 |
+
[dataset_arguments]
|
| 28 |
+
cache_latents_to_disk = true
|
| 29 |
+
debug_dataset = false
|
| 30 |
+
vae_batch_size = 4
|
| 31 |
+
|
| 32 |
+
[training_arguments]
|
| 33 |
+
output_dir = "/content/drive/MyDrive/HSRSplashArtv2/output"
|
| 34 |
+
output_name = "hsrSplashArtv2"
|
| 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 = 10
|
| 42 |
+
max_data_loader_n_workers = 8
|
| 43 |
+
persistent_data_loader_workers = true
|
| 44 |
+
gradient_checkpointing = false
|
| 45 |
+
gradient_accumulation_steps = 1
|
| 46 |
+
mixed_precision = "fp16"
|
| 47 |
+
clip_skip = 2
|
| 48 |
+
lowram = true
|
| 49 |
+
|
| 50 |
+
[logging_arguments]
|
| 51 |
+
log_with = "tensorboard"
|
| 52 |
+
logging_dir = "/content/LoRA/logs"
|
| 53 |
+
log_prefix = "hsrSplashArtv2"
|
| 54 |
+
|
| 55 |
+
[dreambooth_arguments]
|
| 56 |
+
prior_loss_weight = 1.0
|
| 57 |
+
|
| 58 |
+
[saving_arguments]
|
| 59 |
+
save_model_as = "safetensors"
|
| 60 |
+
|
| 61 |
+
[sample_prompt_arguments]
|
| 62 |
+
sample_every_n_epochs = 1
|
| 63 |
+
sample_sampler = "ddim"
|
hsrSplashArtv2_config/dataset_config.toml
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[[datasets]]
|
| 2 |
+
resolution = 768
|
| 3 |
+
min_bucket_reso = 320
|
| 4 |
+
max_bucket_reso = 1280
|
| 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 = ""
|
| 13 |
+
num_repeats = 10
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
[general]
|
| 17 |
+
enable_bucket = true
|
| 18 |
+
caption_extension = ".txt"
|
| 19 |
+
shuffle_caption = true
|
| 20 |
+
keep_tokens = 0
|
| 21 |
+
bucket_reso_steps = 64
|
| 22 |
+
bucket_no_upscale = false
|
hsrSplashArtv2_config/sample_prompt.txt
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
masterpiece, best quality, solo, looking at viewer, short hair, gloves, 1boy, sitting, purple eyes, full body, white hair, male focus, boots, pants, dark skin, black footwear, hair over one eye, animal, dark-skinned male, black background, dog, holding animal, hsr splash art --n (worst quality, low quality:1.4) --w 768 --h 768 --l 7 --s 30
|
| 2 |
+
masterpiece, best quality, 1girl, solo, long hair, looking at viewer, smile, bangs, brown hair, hair ornament, dress, holding, green eyes, jacket, full body, hairclip, black footwear, black background, hsr splash art --n (worst quality, low quality:1.4) --w 768 --h 768 --l 7 --s 30
|
| 3 |
+
masterpiece, best quality, long hair, smile, short hair, open mouth, multiple girls, blonde hair, simple background, brown hair, long sleeves, hat, twintails, blue hair, full body, yellow eyes, braid, boots, 3girls, chibi, star (symbol), scarf, twin braids, coat, fire, >_<, black background, |_|, winter clothes, earmuffs, hsr splash art --n (worst quality, low quality:1.4) --w 768 --h 768 --l 7 --s 30
|
| 4 |
+
masterpiece, best quality, 1girl, solo, long hair, black hair, gloves, dress, holding, full body, weapon, boots, black footwear, holding weapon, moon, bug, black background, butterfly, full moon, scythe, red moon, holding scythe, seele vollerei, hsr splash art --n (worst quality, low quality:1.4) --w 768 --h 768 --l 7 --s 30
|
| 5 |
+
masterpiece, best quality, 1girl, solo, long hair, looking at viewer, skirt, shirt, gloves, long sleeves, holding, purple eyes, jacket, white shirt, purple hair, pantyhose, boots, horns, looking back, black skirt, black footwear, coat, thigh strap, animal, umbrella, cat, sunglasses, holding umbrella, black cat, elite ii (arknights), black umbrella, hsr splash art --n (worst quality, low quality:1.4) --w 768 --h 768 --l 7 --s 30
|