stable-diffusion-inpainting, A-D cup size (4000 steps)

args.json

@@ -1,5 +1,5 @@
 {
-  "pretrained_model_name_or_path": "/home/fant/repos/shiv-diffusers/output/model/3000",
+  "pretrained_model_name_or_path": "runwayml/stable-diffusion-inpainting",
   "pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse",
   "revision": "fp16",
   "tokenizer_name": null,
@@ -7,23 +7,23 @@
   "class_data_dir": null,
   "instance_prompt": null,
   "class_prompt": null,
-  "save_sample_prompt": "woman wearing a blue qsd bikini top with sks pattern|woman wearing a blue qsd bikini bottom with sks pattern|woman wearing a blue zdx bikini top with sks pattern|woman wearing a blue zdx bikini bottom with sks pattern|woman wearing a blue mqw bikini top with sks pattern|woman wearing a blue mqw bikini bottom with sks pattern|woman wearing a blue kcx bikini top with sks pattern|woman wearing a blue kcx bikini bottom with sks pattern|woman wearing a blue qqa bikini top with sks pattern|woman wearing a blue qqa bikini bottom with sks pattern|woman wearing a blue jkp bikini top with sks pattern|woman wearing a blue jkp bikini bottom with sks pattern|woman wearing a blue xxp bikini top with sks pattern|woman wearing a blue xxp bikini bottom with sks pattern",
+  "save_sample_prompt": "close up of woman wearing bikini top, breast size xnjka|close up of woman wearing bikini top, breast size qkdgj|close up of woman wearing bikini top, breast size usmda|close up of woman wearing bikini top, breast size opqng",
   "save_sample_negative_prompt": null,
-  "n_save_sample": 1,
+  "n_save_sample": 4,
   "save_guidance_scale": 7.5,
   "save_infer_steps": 50,
   "pad_tokens": true,
   "with_prior_preservation": true,
   "prior_loss_weight": 1.0,
-  "num_class_images": 300,
-  "output_dir": "/home/fant/repos/shiv-diffusers/output/model-after-3000",
+  "num_class_images": 200,
+  "output_dir": "/home/fant/repos/shiv-diffusers/output/model",
   "seed": 3434554,
   "resolution": 512,
   "center_crop": false,
   "train_text_encoder": true,
   "train_batch_size": 1,
   "sample_batch_size": 4,
-  "num_train_epochs": 5,
+  "num_train_epochs": 25,
   "max_train_steps": 20000,
   "gradient_accumulation_steps": 1,
   "gradient_checkpointing": false,
@@ -42,163 +42,35 @@
   "hub_model_id": null,
   "logging_dir": "logs",
   "log_interval": 10,
-  "save_interval": 3000,
+  "save_interval": 1000,
   "save_min_steps": 0,
   "mixed_precision": "fp16",
   "not_cache_latents": true,
   "local_rank": -1,
   "concepts_list": [
     {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/bali/flower-garden/top/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/bali/flower-garden/top/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/bali/flower-garden/top/2.png"
-      ],
-      "class_prompt": "woman wearing a bikini top",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
-      "instance_prompt": "woman wearing a blue qsd bikini top with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-qsd-bikini-top-with-sks-pattern"
+      "class_prompt": "close up of woman wearing bikini top",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
+      "instance_prompt": "close up of woman wearing bikini top, breast size xnjka",
+      "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/A"
     },
     {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/bali/flower-garden/bottom/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/bali/flower-garden/bottom/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/bali/flower-garden/bottom/2.png"
-      ],
-      "class_prompt": "woman wearing a bikini bottom",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
-      "instance_prompt": "woman wearing a blue qsd bikini bottom with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-qsd-bikini-bottom-with-sks-pattern"
+      "class_prompt": "close up of woman wearing bikini top",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
+      "instance_prompt": "close up of woman wearing bikini top, breast size qkdgj",
+      "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/B"
     },
     {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/pompeio/flower-garden/top/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/pompeio/flower-garden/top/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/pompeio/flower-garden/top/2.png"
-      ],
-      "class_prompt": "woman wearing a bikini top",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
-      "instance_prompt": "woman wearing a blue zdx bikini top with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-zdx-bikini-top-with-sks-pattern"
+      "class_prompt": "close up of woman wearing bikini top",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
+      "instance_prompt": "close up of woman wearing bikini top, breast size usmda",
+      "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/C"
     },
     {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/pompeio/flower-garden/bottom/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/pompeio/flower-garden/bottom/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/pompeio/flower-garden/bottom/2.png"
-      ],
-      "class_prompt": "woman wearing a bikini bottom",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
-      "instance_prompt": "woman wearing a blue zdx bikini bottom with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-zdx-bikini-bottom-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/top/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/top/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/top/2.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/top/3.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/top/4.png"
-      ],
-      "class_prompt": "woman wearing a bikini top",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
-      "instance_prompt": "woman wearing a blue mqw bikini top with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-mqw-bikini-top-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/bottom/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/bottom/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/bottom/2.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/bottom/3.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maria/flower-garden/bottom/4.png"
-      ],
-      "class_prompt": "woman wearing a bikini bottom",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
-      "instance_prompt": "woman wearing a blue mqw bikini bottom with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-mqw-bikini-bottom-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/sky-top/flower-garden/top/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/sky-top/flower-garden/top/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/sky-top/flower-garden/top/2.png"
-      ],
-      "class_prompt": "woman wearing a bikini top",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
-      "instance_prompt": "woman wearing a blue kcx bikini top with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-kcx-bikini-top-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/sky-top/flower-garden/bottom/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/sky-top/flower-garden/bottom/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/sky-top/flower-garden/bottom/2.png"
-      ],
-      "class_prompt": "woman wearing a bikini bottom",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
-      "instance_prompt": "woman wearing a blue kcx bikini bottom with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-kcx-bikini-bottom-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/capri/flower-garden/top/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/capri/flower-garden/top/1.png"
-      ],
-      "class_prompt": "woman wearing a bikini top",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
-      "instance_prompt": "woman wearing a blue qqa bikini top with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-qqa-bikini-top-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/capri/flower-garden/bottom/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/capri/flower-garden/bottom/1.png"
-      ],
-      "class_prompt": "woman wearing a bikini bottom",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
-      "instance_prompt": "woman wearing a blue qqa bikini bottom with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-qqa-bikini-bottom-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/lily-top/flower-garden/top/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/lily-top/flower-garden/top/1.png"
-      ],
-      "class_prompt": "woman wearing a bikini top",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
-      "instance_prompt": "woman wearing a blue jkp bikini top with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-jkp-bikini-top-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/lily-top/flower-garden/bottom/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/lily-top/flower-garden/bottom/1.png"
-      ],
-      "class_prompt": "woman wearing a bikini bottom",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
-      "instance_prompt": "woman wearing a blue jkp bikini bottom with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-jkp-bikini-bottom-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maui/flower-garden/top/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maui/flower-garden/top/1.png"
-      ],
-      "class_prompt": "woman wearing a bikini top",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
-      "instance_prompt": "woman wearing a blue xxp bikini top with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-xxp-bikini-top-with-sks-pattern"
-    },
-    {
-      "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maui/flower-garden/bottom/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/maui/flower-garden/bottom/1.png"
-      ],
-      "class_prompt": "woman wearing a bikini bottom",
-      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
-      "instance_prompt": "woman wearing a blue xxp bikini bottom with sks pattern",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-xxp-bikini-bottom-with-sks-pattern"
+      "class_prompt": "close up of woman wearing bikini top",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
+      "instance_prompt": "close up of woman wearing bikini top, breast size opqng",
+      "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/D"
     }
   ]
 }

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/home/fant/repos/shiv-diffusers/output/model/3000",
+  "_name_or_path": "runwayml/stable-diffusion-inpainting",
   "architectures": [
     "CLIPTextModel"
   ],

text_encoder/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d52abb9e9a9fc012136c660a16e4fc72ac3b6451d2b28f644d3a20361eb330a6
+oid sha256:12ad223d7b4d211daa13f9469ab04d9c7cd1aca37c1267517c848d409ccef736
 size 492308087

tokenizer/tokenizer_config.json

@@ -19,7 +19,7 @@
   },
   "errors": "replace",
   "model_max_length": 77,
-  "name_or_path": "/home/fant/repos/shiv-diffusers/output/model/3000/tokenizer",
+  "name_or_path": "/home/fant/.cache/huggingface/diffusers/models--runwayml--stable-diffusion-inpainting/snapshots/1855202d3acfad9fd7e30e39d06b771d4b859a18/tokenizer",
   "pad_token": "<|endoftext|>",
   "special_tokens_map_file": "./special_tokens_map.json",
   "tokenizer_class": "CLIPTokenizer",

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.7.0.dev0",
-  "_name_or_path": "/home/fant/repos/shiv-diffusers/output/model/3000",
+  "_name_or_path": "runwayml/stable-diffusion-inpainting",
   "act_fn": "silu",
   "attention_head_dim": 8,
   "block_out_channels": [

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0d616067f3edbdc9f2be437e0879569e2ce1180257ead55fbe22f33a9fa5b214
+oid sha256:a4c05032741e2955cc42c7db3732c1fb3a1d22e0d872aa11e59de1398cd48235
 size 3438421925