giers10/skymap-gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add --allow-downloads argument and update model path help text

Browse Source
This commit is contained in:
Victor Giers
2026-05-07 15:13:50 +02:00
parent 62e71b9a94
commit cda671ef1d
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
generate_equirect.py
View File

@@ -795,7 +795,7 @@ def main():
parser.add_argument( parser.add_argument(
'--model-path', '--model-path',
default=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( parser.add_argument(
'--base-model', '--base-model',
@@ -821,11 +821,19 @@ def main():
choices=['euler', 'euler_a', 'heun', 'ddim', 'dpmsolver', 'dpmsolver-sde'], choices=['euler', 'euler_a', 'heun', 'ddim', 'dpmsolver', 'dpmsolver-sde'],
help='Sampler/scheduler override; default uses the pipeline scheduler (Euler for SDXL base)' 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-input', help=argparse.SUPPRESS)
parser.add_argument('--postprocess-output', help=argparse.SUPPRESS) parser.add_argument('--postprocess-output', help=argparse.SUPPRESS)
parser.add_argument('--postprocess-tempdir', help=argparse.SUPPRESS) parser.add_argument('--postprocess-tempdir', help=argparse.SUPPRESS)
args = parser.parse_args() args = parser.parse_args()
if args.allow_downloads:
os.environ["SKYMAP_ALLOW_DOWNLOADS"] = "1"
if args.postprocess_input: if args.postprocess_input:
if not args.postprocess_output or not args.postprocess_tempdir: if not args.postprocess_output or not args.postprocess_tempdir:
parser.error("--postprocess-input requires --postprocess-output and --postprocess-tempdir") parser.error("--postprocess-input requires --postprocess-output and --postprocess-tempdir")