stable-diffusion-inpainting, fine-tuned from breast size, flower-garden (2000 steps)

args.json

@@ -1,27 +1,29 @@
 {
-  "pretrained_model_name_or_path": "runwayml/stable-diffusion-inpainting",
+  "pretrained_model_name_or_path": "davidfant/sd-ecom-brightswimwear",
   "pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse",
-  "revision": "fp16",
+  "revision": "89ed62a56cba5752f773876ef8505a3a2e384808",
   "tokenizer_name": null,
   "instance_data_dir": null,
   "class_data_dir": null,
   "instance_prompt": null,
   "class_prompt": null,
+  "save_sample_prompt": "close up of woman wearing a blue baqsa bikini top with flgap pattern, amfdk breast size|close up of woman wearing a blue baqsa bikini bottom with flgap pattern, amfdk breast size|close up of woman wearing a blue pomqz bikini top with flgap pattern, boqnf breast size|close up of woman wearing a blue pomqz bikini bottom with flgap pattern, boqnf breast size|close up of woman wearing a blue maxfs bikini top with flgap pattern, amfdk breast size|close up of woman wearing a blue maxfs bikini bottom with flgap pattern, amfdk breast size|close up of woman wearing a blue sktqm bikini top with flgap pattern, dpqjd breast size|close up of woman wearing a blue sktqm bikini bottom with flgap pattern, dpqjd breast size|close up of woman wearing a blue capxw bikini top with flgap pattern, dpqjd breast size|close up of woman wearing a blue capxw bikini bottom with flgap pattern, dpqjd breast size|close up of woman wearing a blue litzp bikini top with flgap pattern, amfdk breast size|close up of woman wearing a blue litzp bikini top with flgap pattern, dpqjd breast size|close up of woman wearing a blue litzp bikini bottom with flgap pattern, amfdk breast size|close up of woman wearing a blue litzp bikini bottom with flgap pattern, dpqjd breast size|close up of woman wearing a blue mauqx bikini top with flgap pattern, amfdk breast size|close up of woman wearing a blue mauqx bikini bottom with flgap pattern, amfdk breast size",
   "save_sample_negative_prompt": null,
   "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/model/breast-size",
+  "output_dir": "/home/fant/repos/models/dress_inpainting",
   "seed": 3434554,
   "resolution": 512,
   "center_crop": false,
   "train_text_encoder": true,
   "train_batch_size": 1,
   "sample_batch_size": 4,
-  "num_train_epochs": 27,
+  "num_train_epochs": 7,
   "max_train_steps": 32000,
   "gradient_accumulation_steps": 1,
   "gradient_checkpointing": false,
@@ -44,32 +46,174 @@
   "save_min_steps": 0,
   "mixed_precision": "fp16",
   "not_cache_latents": true,
-  "hflip": false,
   "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": "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, amfdk breast size",
-      "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/0"
+      "instance_prompt": "close up of woman wearing a blue baqsa bikini top with flgap pattern, amfdk breast size",
+      "instance_data_dir": "./data/instance/bali/flower-garden/top"
     },
     {
+      "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": "close up of woman wearing bikini bottom panties",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-bottom-panties",
+      "instance_prompt": "close up of woman wearing a blue baqsa bikini bottom with flgap pattern, amfdk breast size",
+      "instance_data_dir": "./data/instance/bali/flower-garden/bottom-panties"
+    },
+    {
+      "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"
+      ],
+      "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 a blue pomqz bikini top with flgap pattern, boqnf breast size",
+      "instance_data_dir": "./data/instance/pompeio/flower-garden/top"
+    },
+    {
+      "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": "close up of woman wearing bikini bottom panties",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-bottom-panties",
+      "instance_prompt": "close up of woman wearing a blue pomqz bikini bottom with flgap pattern, boqnf breast size",
+      "instance_data_dir": "./data/instance/pompeio/flower-garden/bottom-panties"
+    },
+    {
+      "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": "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 a blue maxfs bikini top with flgap pattern, amfdk breast size",
+      "instance_data_dir": "./data/instance/maria/flower-garden/top"
+    },
+    {
+      "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": "close up of woman wearing bikini bottom panties",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-bottom-panties",
+      "instance_prompt": "close up of woman wearing a blue maxfs bikini bottom with flgap pattern, amfdk breast size",
+      "instance_data_dir": "./data/instance/maria/flower-garden/bottom-panties"
+    },
+    {
+      "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": "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 a blue sktqm bikini top with flgap pattern, dpqjd breast size",
+      "instance_data_dir": "./data/instance/sky-top/flower-garden/top"
+    },
+    {
+      "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": "close up of woman wearing bikini bottom panties",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-bottom-panties",
+      "instance_prompt": "close up of woman wearing a blue sktqm bikini bottom with flgap pattern, dpqjd breast size",
+      "instance_data_dir": "./data/instance/sky-top/flower-garden/bottom-panties"
+    },
+    {
+      "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": "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 a blue capxw bikini top with flgap pattern, dpqjd breast size",
+      "instance_data_dir": "./data/instance/capri/flower-garden/top"
+    },
+    {
+      "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": "close up of woman wearing bikini bottom panties",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-bottom-panties",
+      "instance_prompt": "close up of woman wearing a blue capxw bikini bottom with flgap pattern, dpqjd breast size",
+      "instance_data_dir": "./data/instance/capri/flower-garden/bottom-panties"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/lily-top/flower-garden/top/0.png"
+      ],
       "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, boqnf breast size",
-      "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/1"
+      "instance_prompt": "close up of woman wearing a blue litzp bikini top with flgap pattern, amfdk breast size",
+      "instance_data_dir": "./data/instance/lily-top/flower-garden/top/small-breasts"
     },
     {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/lily-top/flower-garden/top/1.png"
+      ],
       "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, czufm breast size",
-      "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/2"
+      "instance_prompt": "close up of woman wearing a blue litzp bikini top with flgap pattern, dpqjd breast size",
+      "instance_data_dir": "./data/instance/lily-top/flower-garden/top/big-breasts"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/lily-top/flower-garden/bottom/0.png"
+      ],
+      "class_prompt": "close up of woman wearing bikini bottom panties",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-bottom-panties",
+      "instance_prompt": "close up of woman wearing a blue litzp bikini bottom with flgap pattern, amfdk breast size",
+      "instance_data_dir": "./data/instance/lily-top/flower-garden/bottom-panties/small-breasts"
     },
     {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/lily-top/flower-garden/bottom/1.png"
+      ],
+      "class_prompt": "close up of woman wearing bikini bottom panties",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-bottom-panties",
+      "instance_prompt": "close up of woman wearing a blue litzp bikini bottom with flgap pattern, dpqjd breast size",
+      "instance_data_dir": "./data/instance/lily-top/flower-garden/bottom-panties/big-breasts"
+    },
+    {
+      "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": "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, dpqjd breast size",
-      "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/3"
+      "instance_prompt": "close up of woman wearing a blue mauqx bikini top with flgap pattern, amfdk breast size",
+      "instance_data_dir": "./data/instance/maui/flower-garden/top"
+    },
+    {
+      "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": "close up of woman wearing bikini bottom panties",
+      "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-bottom-panties",
+      "instance_prompt": "close up of woman wearing a blue mauqx bikini bottom with flgap pattern, amfdk breast size",
+      "instance_data_dir": "./data/instance/maui/flower-garden/bottom-panties"
     }
   ]
 }