pass correct metadata to network converter
This commit is contained in:
parent
52e78748c8
commit
35c973e55f
|
@ -215,11 +215,11 @@ def convert_model_source(conversion: ConversionContext, model):
|
||||||
logger.info("finished downloading source: %s -> %s", source, dest)
|
logger.info("finished downloading source: %s -> %s", source, dest)
|
||||||
|
|
||||||
|
|
||||||
def convert_model_network(conversion: ConversionContext, network):
|
def convert_model_network(conversion: ConversionContext, model):
|
||||||
format = source_format(network)
|
format = source_format(model)
|
||||||
name = network["name"]
|
name = model["name"]
|
||||||
network_type = network["type"]
|
network_type = model["type"]
|
||||||
source = network["source"]
|
source = model["source"]
|
||||||
|
|
||||||
if network_type == "control":
|
if network_type == "control":
|
||||||
dest = fetch_model(
|
dest = fetch_model(
|
||||||
|
@ -231,12 +231,12 @@ def convert_model_network(conversion: ConversionContext, network):
|
||||||
|
|
||||||
convert_diffusion_control(
|
convert_diffusion_control(
|
||||||
conversion,
|
conversion,
|
||||||
network,
|
model,
|
||||||
dest,
|
dest,
|
||||||
path.join(conversion.model_path, network_type, name),
|
path.join(conversion.model_path, network_type, name),
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
model = network.get("model", None)
|
model = model.get("model", None)
|
||||||
dest = fetch_model(
|
dest = fetch_model(
|
||||||
conversion,
|
conversion,
|
||||||
name,
|
name,
|
||||||
|
@ -421,8 +421,8 @@ def convert_models(conversion: ConversionContext, args, models: Models):
|
||||||
model_errors.append(name)
|
model_errors.append(name)
|
||||||
|
|
||||||
if args.networks and "networks" in models:
|
if args.networks and "networks" in models:
|
||||||
for network in models.get("networks", []):
|
for model in models.get("networks", []):
|
||||||
name = network["name"]
|
name = model["name"]
|
||||||
|
|
||||||
if name in args.skip:
|
if name in args.skip:
|
||||||
logger.info("skipping network: %s", name)
|
logger.info("skipping network: %s", name)
|
||||||
|
|
Loading…
Reference in New Issue