From fd013c88ef6f00ccd65d345c4da181a50c9a1aea Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 23 Feb 2023 18:43:49 -0600 Subject: [PATCH] fix(api): enable optimizations during model conversion --- api/onnx_web/convert/diffusion_stable.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/api/onnx_web/convert/diffusion_stable.py b/api/onnx_web/convert/diffusion_stable.py index c8e6d52e..998e2fb5 100644 --- a/api/onnx_web/convert/diffusion_stable.py +++ b/api/onnx_web/convert/diffusion_stable.py @@ -25,6 +25,8 @@ from diffusers import ( from onnx import load, save_model from torch.onnx import export +from onnx_web.diffusion.load import optimize_pipeline + from ..diffusion.pipeline_onnx_stable_diffusion_upscale import ( OnnxStableDiffusionUpscalePipeline, ) @@ -96,6 +98,8 @@ def convert_diffusion_stable( ).to(ctx.training_device) output_path = Path(dest_path) + optimize_pipeline(ctx, pipeline) + # TEXT ENCODER num_tokens = pipeline.text_encoder.config.max_position_embeddings text_hidden_size = pipeline.text_encoder.config.hidden_size