Fix sample steps

2 files changed, 2 insertions, 2 deletions

diff --git a/dreambooth.py b/dreambooth.py
index b87763e..ec9531e 100644
--- a/dreambooth.py
+++ b/dreambooth.py
@@ -299,7 +299,7 @@ def parse_args():
     parser.add_argument(
         "--sample_steps",
         type=int,
-        default=15,
+        default=20,
         help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.",
     )
     parser.add_argument(
diff --git a/infer.py b/infer.py
index 75c8621..30e11cf 100644
--- a/infer.py
+++ b/infer.py
@@ -38,7 +38,7 @@ default_cmds = {
     "height": 512,
     "batch_size": 1,
     "batch_num": 1,
-    "steps": 30,
+    "steps": 50,
     "guidance_scale": 7.0,
     "seed": None,
     "config": None,