apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

pass correct metadata to network converter

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

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

@ -215,11 +215,11 @@ def convert_model_source(conversion: ConversionContext, model):
logger.info("finished downloading source: %s -> %s", source, dest)
def convert_model_network(conversion: ConversionContext, network):
format = source_format(network)
name = network["name"]
network_type = network["type"]
source = network["source"]
def convert_model_network(conversion: ConversionContext, model):
format = source_format(model)
name = model["name"]
network_type = model["type"]
source = model["source"]
if network_type == "control":
dest = fetch_model(
@ -231,12 +231,12 @@ def convert_model_network(conversion: ConversionContext, network):
convert_diffusion_control(
conversion,
network,
model,
dest,
path.join(conversion.model_path, network_type, name),
)
else:
model = network.get("model", None)
model = model.get("model", None)
dest = fetch_model(
conversion,
name,
@ -421,8 +421,8 @@ def convert_models(conversion: ConversionContext, args, models: Models):
model_errors.append(name)
if args.networks and "networks" in models:
for network in models.get("networks", []):
name = network["name"]
for model in models.get("networks", []):
name = model["name"]
if name in args.skip:
logger.info("skipping network: %s", name)