updated model to new photo

args.json

@@ -7,7 +7,7 @@
   "class_data_dir": null,
   "instance_prompt": null,
   "class_prompt": null,
-  "save_sample_prompt": "<fuenPhoto> person facing the camera",
+  "save_sample_prompt": "<fuenPhoto> animal in the jungle",
   "save_sample_negative_prompt": null,
   "n_save_sample": 4,
   "save_guidance_scale": 7.5,
@@ -16,15 +16,15 @@
   "with_prior_preservation": true,
   "prior_loss_weight": 1.0,
   "num_class_images": 50,
-  "output_dir": "/content/drive/MyDrive/stable_diffusion_weights/photorealistic",
+  "output_dir": "/content/drive/MyDrive/stable_diffusion_weights/photorealistic-v2",
   "seed": 1337,
   "resolution": 512,
   "center_crop": false,
   "train_text_encoder": true,
   "train_batch_size": 1,
-  "sample_batch_size": 4,
-  "num_train_epochs": 40,
-  "max_train_steps": 2000,
+  "sample_batch_size": 6,
+  "num_train_epochs": 20,
+  "max_train_steps": 4000,
   "gradient_accumulation_steps": 1,
   "gradient_checkpointing": false,
   "learning_rate": 1e-06,
@@ -50,10 +50,28 @@
   "local_rank": -1,
   "concepts_list": [
     {
-      "instance_prompt": "<fuenPhoto> portrait of a person",
-      "class_prompt": "photo portrait of a person",
-      "instance_data_dir": "/content/data/training_data",
-      "class_data_dir": "/content/data/photo_portrait_person"
+      "instance_prompt": "<fuenPhoto> person",
+      "class_prompt": "photograph of a person",
+      "instance_data_dir": "/content/data/photo_person_inst",
+      "class_data_dir": "/content/data/photo_person"
+    },
+    {
+      "instance_prompt": "<fuenPhoto> animal",
+      "class_prompt": "photograph of an animal",
+      "instance_data_dir": "/content/data/photo_animal_inst",
+      "class_data_dir": "/content/data/photo_animal"
+    },
+    {
+      "instance_prompt": "<fuenPhoto> red flower",
+      "class_prompt": "photograph of a red flower",
+      "instance_data_dir": "/content/data/photo_red_flower_inst",
+      "class_data_dir": "/content/data/photo_red_flower"
+    },
+    {
+      "instance_prompt": "<fuenPhoto> white flower",
+      "class_prompt": "photograph of a white flower",
+      "instance_data_dir": "/content/data/photo_white_flower_inst",
+      "class_data_dir": "/content/data/photo_white_flower"
     }
   ]
 }

model.ckpt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f203122c85669b807f639c5b9e4d52c870028d90941ec68a9878dfce2da9f878
+oid sha256:83e2bcc72fa0d556bfc21c5608d26bcc0517a118b4f3b9735a3439fdd1d49d1f
 size 2132856622

model_index.json

@@ -2,14 +2,6 @@
   "_class_name": "StableDiffusionPipeline",
   "_diffusers_version": "0.9.0",
   "requires_safety_checker": false,
-  "feature_extractor": [
-    "transformers",
-    "CLIPFeatureExtractor"
-  ],
-  "safety_checker": [
-    "stable_diffusion",
-    "StableDiffusionSafetyChecker"
-  ],
   "scheduler": [
     "diffusers",
     "DDIMScheduler"

text_encoder/pytorch_model.bin

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

unet/diffusion_pytorch_model-003.bin

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4d879746ee9f056086f39fb240875e772e604f7ed4c174dd9b91679c544248eb
-size 3438364325

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4d879746ee9f056086f39fb240875e772e604f7ed4c174dd9b91679c544248eb
+oid sha256:46fcd88e20e9435c964ac218258b358f298e238bf1e09770ab2439e2c632ef80
 size 3438364325