From 5760be710af3c9ce1e2af2ea2d236d45736c340e Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 29 Apr 2023 20:26:01 -0500 Subject: [PATCH] fix(api): restore use of config key when converting from checkpoints --- api/onnx_web/convert/diffusion/diffusers.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/api/onnx_web/convert/diffusion/diffusers.py b/api/onnx_web/convert/diffusion/diffusers.py index 4450712b..84fe8baf 100644 --- a/api/onnx_web/convert/diffusion/diffusers.py +++ b/api/onnx_web/convert/diffusion/diffusers.py @@ -202,11 +202,13 @@ def convert_diffusion_diffusers( single_vae = model.get("single_vae") replace_vae = model.get("vae") pipe_type = model.get("pipeline", "txt2img") + pipe_config = model.get("config", None) device = conversion.training_device dtype = conversion.torch_dtype() logger.debug("using Torch dtype %s for pipeline", dtype) + config_path = path.join(conversion.model_path, "config", pipe_config) dest_path = path.join(conversion.model_path, name) model_index = path.join(dest_path, "model_index.json") model_cnet = path.join(dest_path, "cnet", ONNX_MODEL) @@ -247,6 +249,7 @@ def convert_diffusion_diffusers( logger.debug("loading pipeline from SD checkpoint: %s", source) pipeline = pipe_class.from_ckpt( source, + original_config_file=config_path, torch_dtype=dtype, **pipe_args, ).to(device)