diff options
Diffstat (limited to 'generate.py')
| -rw-r--r-- | generate.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/generate.py b/generate.py index e0d6355..9cbc167 100644 --- a/generate.py +++ b/generate.py @@ -98,6 +98,12 @@ def main(): ) ap.add_argument("prompt", help="Text prompt for the image.") ap.add_argument( + "-m", + "--model", + default=DEFAULT_MODEL, + help=f"Model key: {', '.join(sorted(MODELS))} (default: {DEFAULT_MODEL}).", + ) + ap.add_argument( "-n", "--negative", default="", help="Negative prompt (default: none)." ) ap.add_argument( @@ -149,9 +155,9 @@ def main(): stamp = datetime.now().strftime("%Y%m%d-%H%M%S") out_path = OUT_DIR / f"sdxl-{stamp}.png" - print(f"Loading SDXL base + fp16-fix VAE on XPU ...", file=sys.stderr) + print(f"Loading {args.model} + fp16-fix VAE on XPU ...", file=sys.stderr) t0 = time.time() - pipe = build_pipe() + pipe = build_pipe(args.model) print(f"Loaded in {time.time() - t0:.1f}s", file=sys.stderr) image, elapsed = generate( |