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

ssube · Dec 24, 2023 · 9a077fa · 9a077fa
1 parent a5a60d3
commit 9a077fa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py
@@ -62,8 +62,8 @@
     "img2img": convert_diffusion_diffusers,
     "img2img-sdxl": convert_diffusion_diffusers_xl,
     "inpaint": convert_diffusion_diffusers,
-    "txt2img": convert_diffusion_diffusers,
-    "txt2img-optimum": convert_diffusion_diffusers_optimum,
+    "txt2img": convert_diffusion_diffusers_optimum,
+    "txt2img-legacy": convert_diffusion_diffusers,
     "txt2img-sdxl": convert_diffusion_diffusers_xl,
 }
 

diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py
@@ -818,7 +818,7 @@ def convert_diffusion_diffusers_optimum(
             logger.debug("loading VAE from single tensor file: %s", vae_path)
             pipeline.vae = AutoencoderKL.from_single_file(vae_path)
         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)
 
     if is_torch_2_0: