apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): correctly handle partial extras files

This commit is contained in:
Sean Sube 2023-02-12 12:23:03 -06:00
parent 06d4f32574
commit 9d1c5dca52
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/convert/__main__.py
View File

@ -176,7 +176,7 @@ def fetch_model(
def convert_models(ctx: ConversionContext, args, models: Models):
if args.sources:
if args.sources and "sources" in models:
for model in models.get("sources"):
model = tuple_to_source(model)
name = model.get("name")
@ -189,7 +189,7 @@ def convert_models(ctx: ConversionContext, args, models: Models):
dest = fetch_model(ctx, name, source, model_format=model_format)
logger.info("Finished downloading source: %s -> %s", source, dest)
if args.diffusion:
if args.diffusion and "diffusion" in models:
for model in models.get("diffusion"):
model = tuple_to_diffusion(model)
name = model.get("name")
@ -215,7 +215,7 @@ def convert_models(ctx: ConversionContext, args, models: Models):
source,
)
if args.upscaling:
if args.upscaling and "upscaling" in models:
for model in models.get("upscaling"):
model = tuple_to_upscaling(model)
name = model.get("name")
@ -229,7 +229,7 @@ def convert_models(ctx: ConversionContext, args, models: Models):
)
convert_upscale_resrgan(ctx, model, source)
if args.correction:
if args.correction and "correction" in models:
for model in models.get("correction"):
model = tuple_to_correction(model)
name = model.get("name")