From 054cdc708b2ac5ae3844caa849ae4943717d15fa Mon Sep 17 00:00:00 2001 From: Tianlei Wu Date: Tue, 21 Nov 2023 23:29:27 +0000 Subject: [PATCH] clean output --- .../models/stable_diffusion/demo_txt2img_xl.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_txt2img_xl.py b/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_txt2img_xl.py index 0b9151e2b7e05..7ff1794a68f8c 100644 --- a/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_txt2img_xl.py +++ b/onnxruntime/python/tools/transformers/models/stable_diffusion/demo_txt2img_xl.py @@ -179,9 +179,10 @@ def run_base_and_refiner(warmup=False): if args.nvtx_profile: cudart.cudaProfilerStop() - print("|------------|--------------|") - print("| {:^10} | {:>9.2f} ms |".format("e2e", perf_data["latency"])) - print("|------------|--------------|") + if refiner: + print("|------------|--------------|") + print("| {:^10} | {:>9.2f} ms |".format("e2e", perf_data["latency"])) + print("|------------|--------------|") metadata = get_metadata(args, True) metadata.update({"base." + key: val for key, val in base.metadata().items()})