feat: upload Dreambooth_zwx_woman_v1 checkpoint

Dreambooth_katara_V2_config/config_file.toml

@@ -0,0 +1,50 @@
+[model_arguments]
+v2 = false
+v_parameterization = false
+pretrained_model_name_or_path = "/content/pretrained_model/Realistic_Vision_V2.0.safetensors"
+vae = "/content/vae/stablediffusion.vae.pt"
+
+[optimizer_arguments]
+min_snr_gamma = 5.0
+optimizer_type = "AdamW8bit"
+learning_rate = 1e-6
+max_grad_norm = 1.0
+lr_scheduler = "constant_with_warmup"
+lr_warmup_steps = 5
+
+[dataset_arguments]
+cache_latents = true
+debug_dataset = false
+vae_batch_size = 1
+
+[training_arguments]
+output_dir = "/content/dreambooth/output"
+output_name = "Dreambooth_zwx_woman_v1"
+save_precision = "fp16"
+save_n_epoch_ratio = 1
+save_state = false
+train_batch_size = 2
+max_token_length = 225
+mem_eff_attn = false
+xformers = true
+max_train_steps = 3000
+max_data_loader_n_workers = 8
+persistent_data_loader_workers = true
+seed = 55465
+gradient_checkpointing = false
+gradient_accumulation_steps = 1
+mixed_precision = "fp16"
+clip_skip = 1
+logging_dir = "/content/dreambooth/logs"
+log_prefix = "Dreambooth_zwx_woman_v1"
+noise_offset = 0.05
+
+[sample_prompt_arguments]
+sample_every_n_steps = 100
+sample_sampler = "ddim"
+
+[dreambooth_arguments]
+prior_loss_weight = 1.0
+
+[saving_arguments]
+save_model_as = "safetensors"

Dreambooth_katara_V2_config/dataset_config.toml

@@ -0,0 +1,28 @@
+[[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 = "zwx woman"
+num_repeats = 100
+
+[[datasets.subsets]]
+is_reg = true
+image_dir = "/content/dreambooth/reg_data"
+class_tokens = "zwx woman"
+num_repeats = 1
+
+
+[general]
+enable_bucket = true
+caption_extension = ".caption"
+shuffle_caption = true
+keep_tokens = 1
+bucket_reso_steps = 64
+bucket_no_upscale = false

Dreambooth_katara_V2_config/sample_prompt.txt

@@ -0,0 +1,2 @@
+
+  masterpiece, best quality, 1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt   --n lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry   --w 512   --h 768   --l 7   --s 28