30k step I think

v6_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
+debug_dataset = false
+vae_batch_size = 1
+
+[training_arguments]
+output_dir = "/content/dreambooth/output"
+output_name = "Regainer-v6"
+save_precision = "fp16"
+save_n_epoch_ratio = 1
+save_state = true
+resume = "/content/dreambooth/regainer-v5-state"
+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-v6"
+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"

v6_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

v6_config/sample_prompt.txt

@@ -0,0 +1 @@
+exceptional, best aesthetic, new, newest, best quality, masterpiece, extremely detailed, anime, waifu, 1girl, absurdres, belt, black hair, blue butterfly, bug, butterfly, butterfly hair ornament, butterfly on hand, forehead, hair ornament, haori, highres, insect, japanese clothes, katana, kimetsu no yaiba, kochou shinobu, looking at viewer, parted bangs, parted lips, sheath, sheathed, sidelocks, smile, solo, sword, weapon, white belt,  --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