From da6e813638cf1659b87f60344645d11d7d0eab59 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 24 Dec 2023 06:24:45 -0600 Subject: [PATCH] rename old converter --- api/onnx_web/convert/__main__.py | 8 ++++---- api/onnx_web/convert/diffusion/diffusion.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py index 283afd94..8bd5d1a1 100644 --- a/api/onnx_web/convert/__main__.py +++ b/api/onnx_web/convert/__main__.py @@ -18,7 +18,7 @@ from .client.huggingface import HuggingfaceClient from .correction.gfpgan import convert_correction_gfpgan from .diffusion.control import convert_diffusion_control from .diffusion.diffusion import ( - convert_diffusion_diffusers, + convert_diffusion_diffusers_legacy, convert_diffusion_diffusers_optimum, ) from .diffusion.diffusion_xl import convert_diffusion_diffusers_xl @@ -59,11 +59,11 @@ Models = Dict[str, List[ModelDict]] model_converters: Dict[str, Any] = { "archive": convert_extract_archive, - "img2img": convert_diffusion_diffusers, + "img2img": convert_diffusion_diffusers_optimum, "img2img-sdxl": convert_diffusion_diffusers_xl, - "inpaint": convert_diffusion_diffusers, + "inpaint": convert_diffusion_diffusers_legacy, "txt2img": convert_diffusion_diffusers_optimum, - "txt2img-legacy": convert_diffusion_diffusers, + "txt2img-legacy": convert_diffusion_diffusers_legacy, "txt2img-sdxl": convert_diffusion_diffusers_xl, } diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py index 214d3de5..625e61cb 100644 --- a/api/onnx_web/convert/diffusion/diffusion.py +++ b/api/onnx_web/convert/diffusion/diffusion.py @@ -333,7 +333,7 @@ def collate_cnet(cnet_path): @torch.no_grad() -def convert_diffusion_diffusers( +def convert_diffusion_diffusers_legacy( conversion: ConversionContext, model: Dict, format: Optional[str],