Add --allow-downloads argument and update model path help text
This commit is contained in:
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user