stable-diffusion-inpainting, 7 products (top + bottom) with specific prompt (1000 steps)

args.json

@@ -7,15 +7,15 @@
   "class_data_dir": null,
   "instance_prompt": null,
   "class_prompt": null,
-  "save_sample_prompt": "woman wearing a blue sks bikini top with qsd pattern|woman wearing a blue sks bikini bottom with qsd pattern",
+  "save_sample_prompt": "front of woman wearing a light blue sks bikini top with qsd pattern|back of woman wearing a light blue sks bikini top with qsd pattern|woman standing sideways wearing a light blue sks bikini top with qsd pattern|front of woman wearing a light blue sks bikini bottom with qsd pattern|back of woman wearing a light blue sks bikini bottom with qsd pattern|woman standing sideways wearing a light blue sks bikini bottom with qsd pattern|front of woman wearing a white sks bikini top with nzx pattern|back of woman wearing a white sks bikini top with nzx pattern|woman standing sideways wearing a white sks bikini top with nzx pattern|front of woman wearing a white sks bikini bottom with nzx pattern|back of woman wearing a white sks bikini bottom with nzx pattern|woman standing sideways wearing a white sks bikini bottom with nzx pattern|front of woman wearing a blue sks bikini top with kxq pattern|back of woman wearing a blue sks bikini top with kxq pattern|woman standing sideways wearing a blue sks bikini top with kxq pattern|front of woman wearing a blue sks bikini bottom with kxq pattern|back of woman wearing a blue sks bikini bottom with kxq pattern|woman standing sideways wearing a blue sks bikini bottom with kxq pattern|front of woman wearing  a sks bikini top with khj pattern|back of woman wearing  a sks bikini top with khj pattern|woman standing sideways wearing  a sks bikini top with khj pattern|front of woman wearing  a sks bikini bottom with khj pattern|back of woman wearing  a sks bikini bottom with khj pattern|woman standing sideways wearing  a sks bikini bottom with khj pattern|front of woman wearing a blue sks bikini top with xds pattern|back of woman wearing a blue sks bikini top with xds pattern|woman standing sideways wearing a blue sks bikini top with xds pattern|front of woman wearing a blue sks bikini bottom with xds pattern|back of woman wearing a blue sks bikini bottom with xds pattern|woman standing sideways wearing a blue sks bikini bottom with xds pattern|front of woman wearing a pink sks bikini top with xds pattern|back of woman wearing a pink sks bikini top with xds pattern|woman standing sideways wearing a pink sks bikini top with xds pattern|front of woman wearing a pink sks bikini bottom with xds pattern|back of woman wearing a pink sks bikini bottom with xds pattern|woman standing sideways wearing a pink sks bikini bottom with xds pattern|front of woman wearing a sks bikini top with xyk pattern|back of woman wearing a sks bikini top with xyk pattern|woman standing sideways wearing a sks bikini top with xyk pattern|front of woman wearing a sks bikini bottom with xyk pattern|back of woman wearing a sks bikini bottom with xyk pattern|woman standing sideways wearing a sks bikini bottom with xyk pattern",
   "save_sample_negative_prompt": null,
-  "n_save_sample": 4,
+  "n_save_sample": 1,
   "save_guidance_scale": 7.5,
   "save_infer_steps": 50,
   "pad_tokens": true,
   "with_prior_preservation": true,
   "prior_loss_weight": 1.0,
-  "num_class_images": 100,
+  "num_class_images": 300,
   "output_dir": "/home/fant/repos/shiv-diffusers/output/model",
   "seed": 3434554,
   "resolution": 512,
@@ -23,8 +23,8 @@
   "train_text_encoder": true,
   "train_batch_size": 1,
   "sample_batch_size": 4,
-  "num_train_epochs": 20,
-  "max_train_steps": 4000,
+  "num_train_epochs": 5,
+  "max_train_steps": 20000,
   "gradient_accumulation_steps": 1,
   "gradient_checkpointing": false,
   "learning_rate": 1e-06,
@@ -42,37 +42,165 @@
   "hub_model_id": null,
   "logging_dir": "logs",
   "log_interval": 10,
-  "save_interval": 800,
+  "save_interval": 1000,
   "save_min_steps": 0,
   "mixed_precision": "fp16",
   "not_cache_latents": true,
   "local_rank": -1,
   "concepts_list": [
     {
-      "url": "https://www.brightswimwear.com/products/bali-top-ocean-dream",
-      "instance_prompt": "woman wearing a blue sks bikini top with qsd pattern",
       "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/top/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/top/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/top/2.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/top/3.png"
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/arizona-floral/top/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/arizona-floral/top/1.png"
       ],
       "class_prompt": "woman wearing a bikini top",
       "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-sks-bikini-top-with-qsd-pattern"
+      "instance_prompt": "woman wearing a light blue sks bikini top with qsd pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-light-blue-sks-bikini-top-with-qsd-pattern"
     },
     {
-      "url": "https://www.brightswimwear.com/products/bali-bottom-ocean-dream",
-      "instance_prompt": "woman wearing a blue sks bikini bottom with qsd pattern",
       "image_urls": [
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/bottom/0.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/bottom/1.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/bottom/2.png",
-        "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/bottom/3.png"
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/arizona-floral/bottom/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/arizona-floral/bottom/1.png"
       ],
       "class_prompt": "woman wearing a bikini bottom",
       "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
-      "instance_data_dir": "./data/instance/woman-wearing-a-blue-sks-bikini-bottom-with-qsd-pattern"
+      "instance_prompt": "woman wearing a light blue sks bikini bottom with qsd pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-light-blue-sks-bikini-bottom-with-qsd-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/citrus-floral/top/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/citrus-floral/top/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/citrus-floral/top/2.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/citrus-floral/top/3.png"
+      ],
+      "class_prompt": "woman wearing a bikini top",
+      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
+      "instance_prompt": "woman wearing a white sks bikini top with nzx pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-white-sks-bikini-top-with-nzx-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/citrus-floral/bottom/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/citrus-floral/bottom/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/citrus-floral/bottom/2.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/citrus-floral/bottom/3.png"
+      ],
+      "class_prompt": "woman wearing a bikini bottom",
+      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
+      "instance_prompt": "woman wearing a white sks bikini bottom with nzx pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-white-sks-bikini-bottom-with-nzx-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/flower-garden/top/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/flower-garden/top/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/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 sks bikini top with kxq pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-blue-sks-bikini-top-with-kxq-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/flower-garden/bottom/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/flower-garden/bottom/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/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 sks bikini bottom with kxq pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-blue-sks-bikini-bottom-with-kxq-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/night-floral/top/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/night-floral/top/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/night-floral/top/2.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/night-floral/top/3.png"
+      ],
+      "class_prompt": "woman wearing a bikini top",
+      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
+      "instance_prompt": "woman wearing  a sks bikini top with khj pattern",
+      "instance_data_dir": "./data/instance/woman-wearing--a-sks-bikini-top-with-khj-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/night-floral/bottom/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/night-floral/bottom/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/night-floral/bottom/2.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/night-floral/bottom/3.png"
+      ],
+      "class_prompt": "woman wearing a bikini bottom",
+      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
+      "instance_prompt": "woman wearing  a sks bikini bottom with khj pattern",
+      "instance_data_dir": "./data/instance/woman-wearing--a-sks-bikini-bottom-with-khj-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/ocean-dream/top/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/ocean-dream/top/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/ocean-dream/top/2.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/ocean-dream/top/3.png"
+      ],
+      "class_prompt": "woman wearing a bikini top",
+      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
+      "instance_prompt": "woman wearing a blue sks bikini top with xds pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-blue-sks-bikini-top-with-xds-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/ocean-dream/bottom/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/ocean-dream/bottom/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/ocean-dream/bottom/2.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/ocean-dream/bottom/3.png"
+      ],
+      "class_prompt": "woman wearing a bikini bottom",
+      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
+      "instance_prompt": "woman wearing a blue sks bikini bottom with xds pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-blue-sks-bikini-bottom-with-xds-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/outline-floral/top/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/outline-floral/top/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/outline-floral/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 pink sks bikini top with xds pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-pink-sks-bikini-top-with-xds-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/outline-floral/bottom/0.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/outline-floral/bottom/1.png",
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/outline-floral/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 pink sks bikini bottom with xds pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-pink-sks-bikini-bottom-with-xds-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/sweet-vacay/top/0.png"
+      ],
+      "class_prompt": "woman wearing a bikini top",
+      "class_data_dir": "./data/class/woman-wearing-a-bikini-top",
+      "instance_prompt": "woman wearing a sks bikini top with xyk pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-sks-bikini-top-with-xyk-pattern"
+    },
+    {
+      "image_urls": [
+        "https://storage.googleapis.com/dreambooth-training/bikini-crops--40-padding/sweet-vacay/bottom/0.png"
+      ],
+      "class_prompt": "woman wearing a bikini bottom",
+      "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom",
+      "instance_prompt": "woman wearing a sks bikini bottom with xyk pattern",
+      "instance_data_dir": "./data/instance/woman-wearing-a-sks-bikini-bottom-with-xyk-pattern"
     }
   ]
 }

text_encoder/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4ed42652dea812c9e5958f6dd516d0d6f5df7eaca164a6f22ca6ad566b4f52e9
+oid sha256:a05b4cbfbe53f7d5b0797f547fb3e22167a236786898fa25adfa002e0e063638
 size 492308087

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:87069178b258d79ef70fb28157b4f51e0fd7b61ce486e069f16c5ca6661302af
+oid sha256:04a6b294cd834ddf0c01c58107b78dfd5581f2004036468fe6b2db36b2a6685a
 size 3438421925