diff options
| author | Volpeon <git@volpeon.ink> | 2022-12-06 08:54:12 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-12-06 08:54:12 +0100 |
| commit | 634a53304333d66c94bb207b9f003e8df6dceb41 (patch) | |
| tree | 6c177d0066aa4a5a7395e7f8a21d53558f6c969b | |
| parent | Update (diff) | |
| download | textual-inversion-diff-634a53304333d66c94bb207b9f003e8df6dceb41.tar.gz textual-inversion-diff-634a53304333d66c94bb207b9f003e8df6dceb41.tar.bz2 textual-inversion-diff-634a53304333d66c94bb207b9f003e8df6dceb41.zip | |
Fix sample steps
| -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, |