apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): replace legacy SD converter with optimum-based one by default

This commit is contained in:
Sean Sube 2023-12-23 22:55:52 -06:00
parent a5a60d3e15
commit 9a077fae15
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -62,8 +62,8 @@ model_converters: Dict[str, Any] = {
"img2img": convert_diffusion_diffusers, "img2img": convert_diffusion_diffusers,
"img2img-sdxl": convert_diffusion_diffusers_xl, "img2img-sdxl": convert_diffusion_diffusers_xl,
"inpaint": convert_diffusion_diffusers, "inpaint": convert_diffusion_diffusers,
"txt2img": convert_diffusion_diffusers, "txt2img": convert_diffusion_diffusers_optimum,
"txt2img-optimum": convert_diffusion_diffusers_optimum, "txt2img-legacy": convert_diffusion_diffusers,
"txt2img-sdxl": convert_diffusion_diffusers_xl, "txt2img-sdxl": convert_diffusion_diffusers_xl,
} }

2
api/onnx_web/convert/diffusion/diffusion.py
View File

@ -818,7 +818,7 @@ def convert_diffusion_diffusers_optimum(
logger.debug("loading VAE from single tensor file: %s", vae_path) logger.debug("loading VAE from single tensor file: %s", vae_path)
pipeline.vae = AutoencoderKL.from_single_file(vae_path) pipeline.vae = AutoencoderKL.from_single_file(vae_path)
else: else:
logger.debug("loading VAE from single tensor file: %s", vae_path) logger.debug("loading pretrained VAE from path: %s", replace_vae)
pipeline.vae = AutoencoderKL.from_pretrained(replace_vae) pipeline.vae = AutoencoderKL.from_pretrained(replace_vae)
if is_torch_2_0: if is_torch_2_0: