feat: upload OCv3 lora model
Browse files- OCv3_config/config_file.toml +63 -0
- OCv3_config/dataset_config.toml +15 -0
- OCv3_config/sample_prompt.toml +12 -0
OCv3_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/AnyLoRA_noVae_fp16-pruned.ckpt"
|
| 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 = 64
|
| 12 |
+
network_alpha = 32
|
| 13 |
+
network_args = []
|
| 14 |
+
network_train_unet_only = false
|
| 15 |
+
network_train_text_encoder_only = false
|
| 16 |
+
|
| 17 |
+
[optimizer_arguments]
|
| 18 |
+
min_snr_gamma = 1.0
|
| 19 |
+
optimizer_type = "AdamW8bit"
|
| 20 |
+
learning_rate = 0.0001
|
| 21 |
+
max_grad_norm = 1.0
|
| 22 |
+
optimizer_args = []
|
| 23 |
+
lr_scheduler = "constant_with_warmup"
|
| 24 |
+
lr_warmup_steps = 135
|
| 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 = "OCv3"
|
| 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 = 20
|
| 42 |
+
max_data_loader_n_workers = 8
|
| 43 |
+
persistent_data_loader_workers = true
|
| 44 |
+
seed = 46
|
| 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 = "OCv3"
|
| 54 |
+
|
| 55 |
+
[sample_prompt_arguments]
|
| 56 |
+
sample_every_n_epochs = 1
|
| 57 |
+
sample_sampler = "k_dpm_2_a"
|
| 58 |
+
|
| 59 |
+
[dreambooth_arguments]
|
| 60 |
+
prior_loss_weight = 1.0
|
| 61 |
+
|
| 62 |
+
[saving_arguments]
|
| 63 |
+
save_model_as = "safetensors"
|
OCv3_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 = 5
|
| 9 |
+
|
| 10 |
+
|
| 11 |
+
[general]
|
| 12 |
+
enable_bucket = true
|
| 13 |
+
caption_extension = ".txt"
|
| 14 |
+
shuffle_caption = true
|
| 15 |
+
keep_tokens = 0
|
OCv3_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, breasts, looking at viewer, blush, skirt, large breasts, simple background, shirt, long sleeves, white background, bow, holding, sitting, school uniform, white shirt, one eye closed, shoes, collared shirt, bowtie, black skirt, black footwear, red bow, book, chair, red bowtie, holding book"
|
| 9 |
+
|
| 10 |
+
[[prompt.subset]]
|
| 11 |
+
prompt = "masterpiece, best quality, 1girl, solo, looking at viewer, blush, simple background, white background, dress, bow, hair between eyes, brown eyes, standing, full body, short sleeves, boots, parted lips, one eye closed, shoes, puffy sleeves, sailor collar, white dress, puffy short sleeves, black bow, brown footwear, sailor dress, black sailor collar, pleated dress"
|
| 12 |
+
|