From 52e78748c874ed11c27010b2c3f2248a062fae48 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 9 Dec 2023 23:05:41 -0600 Subject: [PATCH] keep model field optional --- api/onnx_web/convert/__main__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py index f3c02953..eefa6052 100644 --- a/api/onnx_web/convert/__main__.py +++ b/api/onnx_web/convert/__main__.py @@ -218,7 +218,6 @@ def convert_model_source(conversion: ConversionContext, model): def convert_model_network(conversion: ConversionContext, network): format = source_format(network) name = network["name"] - model = network["model"] network_type = network["type"] source = network["source"] @@ -237,6 +236,7 @@ def convert_model_network(conversion: ConversionContext, network): path.join(conversion.model_path, network_type, name), ) else: + model = network.get("model", None) dest = fetch_model( conversion, name,