From cda671ef1d2c4a6360a7ee2062ff38e0c38741f0 Mon Sep 17 00:00:00 2001 From: Victor Giers Date: Thu, 7 May 2026 15:13:50 +0200 Subject: [PATCH] Add --allow-downloads argument and update model path help text --- generate_equirect.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/generate_equirect.py b/generate_equirect.py index 515e52c..f30422e 100644 --- a/generate_equirect.py +++ b/generate_equirect.py @@ -795,7 +795,7 @@ def main(): parser.add_argument( '--model-path', default=MODEL_PATH, - help='Diffusers model id or local path (default: ProGamerGov/sdxl-360-diffusion)' + help='Diffusers model id or local path (default: local models/sdxl360Diffusion_v10.safetensors when present)' ) parser.add_argument( '--base-model', @@ -821,11 +821,19 @@ 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( + '--allow-downloads', + action='store_true', + help='Allow missing model files to be downloaded from remote model hubs' + ) 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.allow_downloads: + os.environ["SKYMAP_ALLOW_DOWNLOADS"] = "1" + if args.postprocess_input: if not args.postprocess_output or not args.postprocess_tempdir: parser.error("--postprocess-input requires --postprocess-output and --postprocess-tempdir")