Upload 3 files

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-v4-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"

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

config/sample_prompt.txt

@@ -0,0 +1 @@
+exceptional, best aesthetic, new, newest, best quality, masterpiece, extremely detailed, anime, waifu, 1girl, absurdres, alternate costume, apron, black footwear, black thighhighs, blonde hair, blush, breasts, commentary, enmaided, flower, foot out of frame, genshin impact, hair flower, hair ornament, highres, large breasts, legs, looking at viewer, lumine (genshin impact), maid, maid headdress, medium hair, parted lips, short sleeves, simple background, solo, sweat, symbol-only commentary, thighhighs, thighs, twitter username, white background, white headwear, yellow eyes,  --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