diff options
| -rw-r--r-- | dreambooth.py | 2 | ||||
| -rw-r--r-- | 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(): | |||
| 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( |
| @@ -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, |
