update README

CompVis · MushroomFleet · Sep 27, 2022 · Oct 8, 2022 · Oct 8, 2022 · Oct 8, 2022
commit d8f37eb54e13aa69eafef3aafd1c7a1895de7dda
diff --git a/README.md b/README.md
diff --git a/assets/grid-0033.png b/assets/grid-0033.png
diff --git a/assets/grid-0035.png b/assets/grid-0035.png
diff --git a/assets/grid-0043.png b/assets/grid-0043.png
diff --git a/assets/grid-0047.png b/assets/grid-0047.png
diff --git a/assets/grid-0072.png b/assets/grid-0072.png
diff --git a/assets/grid-0073.png b/assets/grid-0073.png
diff --git a/assets/grid-0075.png b/assets/grid-0075.png
diff --git a/assets/grid-0089.png b/assets/grid-0089.png
diff --git a/assets/grid-0091.png b/assets/grid-0091.png
diff --git a/assets/grid-0096.png b/assets/grid-0096.png
diff --git a/assets/grid-0097.png b/assets/grid-0097.png
diff --git a/assets/grid-0131.png b/assets/grid-0131.png
diff --git a/assets/grid-0133.png b/assets/grid-0133.png
diff --git a/assets/grid-0135.png b/assets/grid-0135.png
diff --git a/scripts/txt2img.py b/scripts/txt2img.py
@@ -236,7 +236,7 @@ def main():
         "--aesthetic_steps",
         type=int,
         help="number of steps for the aesthetic personalization",
-        default=5,
+        default=10,
     )
     parser.add_argument(
         "--aesthetic_lr",
@@ -348,7 +348,8 @@ def main():
                             x_samples_ddim.cpu().permute(0, 2, 3, 1).numpy()
                         )
 
-                        x_checked_image, has_nsfw_concept = check_safety(x_samples_ddim)
+                        # x_checked_image, has_nsfw_concept = check_safety(x_samples_ddim)
+                        x_checked_image = x_samples_ddim
 
                         x_checked_image_torch = torch.from_numpy(
                             x_checked_image