feat: upload OCv2 lora model
Browse files- OCv2_config/config_file.toml +63 -0
- OCv2_config/dataset_config.toml +15 -0
- OCv2_config/sample_prompt.toml +12 -0
OCv2_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 = 35
|
| 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 = "OCv2"
|
| 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 = 5
|
| 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 = "OCv2"
|
| 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"
|
OCv2_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 = 50
|
| 9 |
+
|
| 10 |
+
|
| 11 |
+
[general]
|
| 12 |
+
enable_bucket = true
|
| 13 |
+
caption_extension = ".txt"
|
| 14 |
+
shuffle_caption = true
|
| 15 |
+
keep_tokens = 0
|
OCv2_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, simple background, long sleeves, white background, navel, medium breasts, nipples, standing, collarbone, full body, boots, one eye closed, pussy, coat, sleeves past wrists, grey eyes, brown footwear, thigh gap, cleft of venus, naked coat, naked jacket"
|
| 9 |
+
|
| 10 |
+
[[prompt.subset]]
|
| 11 |
+
prompt = "masterpiece, best quality, 1girl, solo, breasts, looking at viewer, blush, simple background, white background, dress, holding, standing, collarbone, full body, flower, short sleeves, parted lips, one eye closed, puffy sleeves, black footwear, white dress, puffy short sleeves, grey eyes, rose, pink flower, bouquet, wedding dress, pink rose, holding bouquet"
|
| 12 |
+
|