From 984dd02df32927e457b4b0d117ac9d86783edd60 Mon Sep 17 00:00:00 2001 From: Chester Liu Date: Sat, 13 May 2023 11:43:29 +0800 Subject: [PATCH] Update optimize_pipeline.py to use __name__ detection (#15866) ### Description Use `__name__` detection in `optimize_pipeline.py`. ### Motivation and Context It prevents unwanted execution of `main` when importing the file. --- .../transformers/models/stable_diffusion/optimize_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 5e8a25eb0b..f1b60ba1d2 100644 --- a/onnxruntime/python/tools/transformers/models/stable_diffusion/optimize_pipeline.py +++ b/onnxruntime/python/tools/transformers/models/stable_diffusion/optimize_pipeline.py @@ -312,4 +312,5 @@ def main(): ) -main() +if __name__ == "__main__": + main()