From 51a9ae32be6091ff9a5b77e3e8cd6bdf267bdeb5 Mon Sep 17 00:00:00 2001 From: Dilip Parasu Date: Wed, 13 Sep 2023 11:02:59 +0530 Subject: [PATCH] Update optimize_pipeline.py Fixed the import statements causing the script not to run in standalone mode --- .../models/stable_diffusion/optimize_pipeline.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/onnxruntime/python/tools/transformers/models/stable_diffusion/optimize_pipeline.py b/onnxruntime/python/tools/transformers/models/stable_diffusion/optimize_pipeline.py index 22fee4bfeab29..086d4ca046e12 100644 --- a/onnxruntime/python/tools/transformers/models/stable_diffusion/optimize_pipeline.py +++ b/onnxruntime/python/tools/transformers/models/stable_diffusion/optimize_pipeline.py @@ -29,11 +29,11 @@ import coloredlogs import onnx -from fusion_options import FusionOptions -from onnx_model_clip import ClipOnnxModel -from onnx_model_unet import UnetOnnxModel -from onnx_model_vae import VaeOnnxModel -from optimizer import optimize_by_onnxruntime, optimize_model +from onnxruntime.transformers.fusion_options import FusionOptions +from onnxruntime.transformers.onnx_model_clip import ClipOnnxModel +from onnxruntime.transformers.onnx_model_unet import UnetOnnxModel +from onnxruntime.transformers.onnx_model_vae import VaeOnnxModel +from onnxruntime.transformers.optimizer import optimize_by_onnxruntime, optimize_model from packaging import version import onnxruntime