From 634a53304333d66c94bb207b9f003e8df6dceb41 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Tue, 6 Dec 2022 08:54:12 +0100 Subject: Fix sample steps --- dreambooth.py | 2 +- infer.py | 2 +- 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, -- cgit v1.2.3-54-g00ecf