From c445ab9b12194495f80cc6759233d3e57a409577 Mon Sep 17 00:00:00 2001 From: Victor Giers Date: Thu, 7 May 2026 12:03:54 +0200 Subject: [PATCH] auto-git: [change] generate_equirect.py --- generate_equirect.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/generate_equirect.py b/generate_equirect.py index 17e651e..b43d1b2 100644 --- a/generate_equirect.py +++ b/generate_equirect.py @@ -720,8 +720,34 @@ def main(): choices=['euler', 'euler_a', 'heun', 'ddim', 'dpmsolver', 'dpmsolver-sde'], help='Sampler/scheduler override; default uses the pipeline scheduler (Euler for SDXL base)' ) + parser.add_argument('--postprocess-input', help=argparse.SUPPRESS) + parser.add_argument('--postprocess-output', help=argparse.SUPPRESS) + parser.add_argument('--postprocess-tempdir', help=argparse.SUPPRESS) args = parser.parse_args() + if args.postprocess_input: + if not args.postprocess_output or not args.postprocess_tempdir: + parser.error("--postprocess-input requires --postprocess-output and --postprocess-tempdir") + try: + result_path = postprocess_image( + args.prompt, + os.path.abspath(args.postprocess_input), + os.path.abspath(args.postprocess_output), + os.path.abspath(args.postprocess_tempdir), + upscale=args.upscale, + steps=args.steps, + guidance=args.guidance, + width=args.width, + height=args.height, + seam_inpaint=args.seam_inpaint, + ) + print(result_path) + shutil.rmtree(args.postprocess_tempdir, ignore_errors=True) + return + except Exception as e: # noqa: BLE001 + print(f"Generation failed: {e}") + raise + base = sanitize_name(args.prompt) target = args.output or next_filename(args.output_dir, base, args.width, args.height) output_abs = os.path.abspath(target)