From d34cc72ceedf964da446ea3749ba43aaaeac7b22 Mon Sep 17 00:00:00 2001 From: "Danilo M." Date: Wed, 8 Jul 2026 15:53:29 +0200 Subject: Add -m/--model flag to one-shot CLI Co-Authored-By: Claude Opus 4.8 --- generate.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'generate.py') diff --git a/generate.py b/generate.py index e0d6355..9cbc167 100644 --- a/generate.py +++ b/generate.py @@ -97,6 +97,12 @@ def main(): description="Generate a single SDXL-base image locally on Intel Arc (XPU)." ) 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)." ) @@ -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( -- cgit v1.2.3