feat: upload annahighjumper_zwx_v1 lora model
Browse files
annahighjumper_zwx_v1_config/config_file.toml
ADDED
|
@@ -0,0 +1,59 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[model_arguments]
|
| 2 |
+
v2 = false
|
| 3 |
+
v_parameterization = false
|
| 4 |
+
pretrained_model_name_or_path = "/content/pretrained_model/AnyLoRA.safetensors"
|
| 5 |
+
|
| 6 |
+
[additional_network_arguments]
|
| 7 |
+
no_metadata = false
|
| 8 |
+
unet_lr = 0.0001
|
| 9 |
+
text_encoder_lr = 5e-5
|
| 10 |
+
network_module = "lycoris.kohya"
|
| 11 |
+
network_dim = 16
|
| 12 |
+
network_alpha = 8
|
| 13 |
+
network_args = [ "conv_dim=8", "conv_alpha=1", "algo=lora",]
|
| 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 |
+
lr_scheduler = "constant"
|
| 23 |
+
lr_warmup_steps = 0
|
| 24 |
+
|
| 25 |
+
[dataset_arguments]
|
| 26 |
+
cache_latents = true
|
| 27 |
+
debug_dataset = false
|
| 28 |
+
vae_batch_size = 4
|
| 29 |
+
|
| 30 |
+
[training_arguments]
|
| 31 |
+
output_dir = "/content/LoRA/output"
|
| 32 |
+
output_name = "annahighjumper_zwx_v1"
|
| 33 |
+
save_precision = "fp16"
|
| 34 |
+
save_every_n_epochs = 1
|
| 35 |
+
train_batch_size = 6
|
| 36 |
+
max_token_length = 225
|
| 37 |
+
mem_eff_attn = false
|
| 38 |
+
xformers = true
|
| 39 |
+
max_train_epochs = 8
|
| 40 |
+
max_data_loader_n_workers = 8
|
| 41 |
+
persistent_data_loader_workers = true
|
| 42 |
+
gradient_checkpointing = false
|
| 43 |
+
gradient_accumulation_steps = 1
|
| 44 |
+
mixed_precision = "fp16"
|
| 45 |
+
clip_skip = 2
|
| 46 |
+
logging_dir = "/content/LoRA/logs"
|
| 47 |
+
log_prefix = "annahighjumper_zwx_v1"
|
| 48 |
+
noise_offset = 0.02
|
| 49 |
+
lowram = true
|
| 50 |
+
|
| 51 |
+
[sample_prompt_arguments]
|
| 52 |
+
sample_every_n_epochs = 1
|
| 53 |
+
sample_sampler = "ddim"
|
| 54 |
+
|
| 55 |
+
[dreambooth_arguments]
|
| 56 |
+
prior_loss_weight = 1.0
|
| 57 |
+
|
| 58 |
+
[saving_arguments]
|
| 59 |
+
save_model_as = "safetensors"
|
annahighjumper_zwx_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 = "zwx anna"
|
| 13 |
+
num_repeats = 25
|
| 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
|
annahighjumper_zwx_v1_config/sample_prompt.txt
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
zwxanna, masterpiece, best quality, 1girl, aqua eyes, pony tail, white hair, baseball cap, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt --n 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 --w 512 --h 768 --l 7 --s 28
|