apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): make diffusion model config optional again

This commit is contained in:
Sean Sube 2023-02-13 22:49:50 -06:00
parent 6b6f2f0eff
commit 4cc1d63cec
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/convert/diffusion_original.py
View File

@ -1409,7 +1409,6 @@ def convert_diffusion_original(
model: ModelDict,
source: str,
):
config = model["config"]
name = model["name"]
source = source or model["source"]
@ -1428,7 +1427,7 @@ def convert_diffusion_original(
logger.info("Torch pipeline already exists, reusing: %s", torch_path)
else:
logger.info("Converting original Diffusers check to Torch model: %s -> %s", source, torch_path)
extract_checkpoint(ctx, torch_name, source, config_file=config)
extract_checkpoint(ctx, torch_name, source, config_file=model.get("config"))
logger.info("Converted original Diffusers checkpoint to Torch model.")
convert_diffusion_stable(ctx, model, working_name)