diff options
| author | Danilo M. <danix@danix.xyz> | 2026-07-08 15:53:29 +0200 |
|---|---|---|
| committer | Danilo M. <danix@danix.xyz> | 2026-07-08 15:53:29 +0200 |
| commit | d34cc72ceedf964da446ea3749ba43aaaeac7b22 (patch) | |
| tree | ce67637a9ef939f0f0c8f24648fad704cb5df24c | |
| parent | 4d137f35e724e98128916709c9fcf154190c744c (diff) | |
| download | imggen-d34cc72ceedf964da446ea3749ba43aaaeac7b22.tar.gz imggen-d34cc72ceedf964da446ea3749ba43aaaeac7b22.zip | |
Add -m/--model flag to one-shot CLI
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
| -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( |