apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

register optimum-based SD converter

This commit is contained in:
Sean Sube 2023-12-23 22:18:09 -06:00
parent 5447d8c159
commit 14208de393
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -59,6 +59,7 @@ CONVERT_PIPELINES = {
"panorama": StableDiffusionPipeline, "panorama": StableDiffusionPipeline,
"pix2pix": StableDiffusionInstructPix2PixPipeline, "pix2pix": StableDiffusionInstructPix2PixPipeline,
"txt2img": StableDiffusionPipeline, "txt2img": StableDiffusionPipeline,
"txt2img-optimum": StableDiffusionPipeline,
"upscale": StableDiffusionUpscalePipeline, "upscale": StableDiffusionUpscalePipeline,
} }
@ -781,7 +782,10 @@ def convert_diffusion_diffusers_optimum(
# diffusers go into a directory rather than .onnx file # diffusers go into a directory rather than .onnx file
logger.info( logger.info(
"converting Stable Diffusion model %s: %s -> %s/", name, source, dest_path "converting Stable Diffusion model %s using Optimum: %s -> %s/",
name,
source,
dest_path,
) )
if path.exists(dest_path) and path.exists(model_index): if path.exists(dest_path) and path.exists(model_index):