diff options
| author | C43H66N12O12S2 <36072735+C43H66N12O12S2@users.noreply.github.com> | 2022-09-19 19:36:42 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-09-19 22:49:53 +0300 |
| commit | 53651696dc1492f3b7cc009d64a55532bc787aa7 (patch) | |
| tree | b804d6fd21c987927b0ebd80d6e2e2b065f816e1 /modules | |
| parent | 942acd4e1238ec881f3492798b4cd8eca1d0cf8d (diff) | |
avoid exception if empty
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/shared.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/shared.py b/modules/shared.py index b6a3fd5b..38f54a80 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -22,7 +22,7 @@ parser.add_argument("--config", type=str, default=os.path.join(sd_path, "configs parser.add_argument("--ckpt", type=str, default=sd_model_file, help="path to checkpoint of stable diffusion model; this checkpoint will be added to the list of checkpoints and loaded by default if you don't have a checkpoint selected in settings",)
parser.add_argument("--ckpt-dir", type=str, default=os.path.join(script_path, 'models'), help="path to directory with stable diffusion checkpoints",)
parser.add_argument("--gfpgan-dir", type=str, help="GFPGAN directory", default=('./src/gfpgan' if os.path.exists('./src/gfpgan') else './GFPGAN'))
-parser.add_argument("--gfpgan-model", type=str, help="GFPGAN model file name", default=next(iter(glob('GFPGAN*.pth'))))
+parser.add_argument("--gfpgan-model", type=str, help="GFPGAN model file name", default=next(iter(glob('GFPGAN*.pth')), ''))
parser.add_argument("--no-half", action='store_true', help="do not switch the model to 16-bit floats")
parser.add_argument("--no-progressbar-hiding", action='store_true', help="do not hide progressbar in gradio UI (we hide it because it slows down ML if you have hardware acceleration in browser)")
parser.add_argument("--max-batch-count", type=int, default=16, help="maximum batch count value for the UI")
|