apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

keep model field optional

This commit is contained in:
Sean Sube 2023-12-09 23:05:41 -06:00
parent e052578a20
commit 52e78748c8
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -218,7 +218,6 @@ def convert_model_source(conversion: ConversionContext, model):
def convert_model_network(conversion: ConversionContext, network): def convert_model_network(conversion: ConversionContext, network):
format = source_format(network) format = source_format(network)
name = network["name"] name = network["name"]
model = network["model"]
network_type = network["type"] network_type = network["type"]
source = network["source"] source = network["source"]
@ -237,6 +236,7 @@ def convert_model_network(conversion: ConversionContext, network):
path.join(conversion.model_path, network_type, name), path.join(conversion.model_path, network_type, name),
) )
else: else:
model = network.get("model", None)
dest = fetch_model( dest = fetch_model(
conversion, conversion,
name, name,