feat: upload Regainer-v4 checkpoint

v4_config/config_file.toml

@@ -0,0 +1,48 @@
+[model_arguments]
+v2 = true
+v_parameterization = true
+pretrained_model_name_or_path = "/content/pretrained_model/Regainer-v3.safetensors"
+
+[optimizer_arguments]
+min_snr_gamma = 5
+optimizer_type = "AdamW8bit"
+learning_rate = 2e-6
+max_grad_norm = 1.0
+lr_scheduler = "constant"
+lr_warmup_steps = 0
+
+[dataset_arguments]
+cache_latents = true
+cache_latents_to_disk = true
+debug_dataset = false
+vae_batch_size = 1
+
+[training_arguments]
+output_dir = "/content/dreambooth/output"
+output_name = "Regainer-v4"
+save_precision = "fp16"
+save_n_epoch_ratio = 1
+save_state = true
+train_batch_size = 2
+max_token_length = 225
+mem_eff_attn = false
+xformers = true
+max_train_steps = 7292
+max_data_loader_n_workers = 8
+persistent_data_loader_workers = true
+gradient_checkpointing = false
+gradient_accumulation_steps = 1
+mixed_precision = "fp16"
+logging_dir = "/content/dreambooth/logs"
+log_prefix = "Regainer-v4"
+noise_offset = 0.1
+
+[sample_prompt_arguments]
+sample_every_n_steps = 200
+sample_sampler = "ddim"
+
+[dreambooth_arguments]
+prior_loss_weight = 1.0
+
+[saving_arguments]
+save_model_as = "safetensors"

v4_config/dataset_config.toml

@@ -0,0 +1,22 @@
+[[datasets]]
+resolution = 512
+min_bucket_reso = 256
+max_bucket_reso = 1024
+caption_dropout_rate = 0
+caption_tag_dropout_rate = 0
+caption_dropout_every_n_epochs = 0
+flip_aug = false
+color_aug = false
+[[datasets.subsets]]
+image_dir = "/content/dreambooth/train_data"
+class_tokens = "mksks style"
+num_repeats = 2
+
+
+[general]
+enable_bucket = true
+caption_extension = ".txt"
+shuffle_caption = true
+keep_tokens = 0
+bucket_reso_steps = 64
+bucket_no_upscale = false

v4_config/sample_prompt.txt

@@ -0,0 +1 @@
+exceptional, best aesthetic, new, newest, best quality, masterpiece, extremely detailed, anime, waifu, 1girl, bag, black jacket, black pantyhose, blurry, blurry background, blush, braid, breasts, brown eyes, brown hair, cellphone, closed mouth, commentary, depth of field, earrings, frown, grey skirt, hands up, highres, holding, holding pen, indoors, jacket, jewelry, kanojo okarishimasu, large breasts, long hair, long sleeves, looking at viewer, mizuhara chizuru, necklace, one side up, pantyhose, pearl necklace, pen, phone, red sweater, ribbed sweater, sitting, skirt, sleeves past wrists, smartphone, solo, stud earrings, sweater, table, window,     --n simple background, white background, lowres, bad anatomy, bad hands, missing finger, extra digits, fewer digits, blurry, mutated hands and fingers, poorly drawn face, mutation, deformed face, ugly, bad proportions, extra limbs, extra face, double head, extra head, extra feet, monster, logo, cropped, worst quality, jpeg, humpbacked, long body, long neck, jpeg artifacts, deleted, old, oldest, censored,bad aesthetic, mosaic censoring, bar censor, blur censor, simple background   --w 512   --h 768   --l 7   --s 28