apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): restore use of config key when converting from checkpoints

This commit is contained in:
Sean Sube 2023-04-29 20:26:01 -05:00
parent 4ea372d53a
commit 5760be710a
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/convert/diffusion/diffusers.py
View File

@ -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)