summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--dreambooth.py2
-rw-r--r--infer.py2
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():
299 parser.add_argument( 299 parser.add_argument(
300 "--sample_steps", 300 "--sample_steps",
301 type=int, 301 type=int,
302 default=15, 302 default=20,
303 help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.", 303 help="Number of steps for sample generation. Higher values will result in more detailed samples, but longer runtimes.",
304 ) 304 )
305 parser.add_argument( 305 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 = {
38 "height": 512, 38 "height": 512,
39 "batch_size": 1, 39 "batch_size": 1,
40 "batch_num": 1, 40 "batch_num": 1,
41 "steps": 30, 41 "steps": 50,
42 "guidance_scale": 7.0, 42 "guidance_scale": 7.0,
43 "seed": None, 43 "seed": None,
44 "config": None, 44 "config": None,