From 81d6e8c1960a225426b97498e72ab093b87f3329 Mon Sep 17 00:00:00 2001 From: Yi Zhang Date: Thu, 27 Jun 2024 14:47:46 +0800 Subject: [PATCH] roll back change --- .../python/tools/transformers/models/llama/convert_to_onnx.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/onnxruntime/python/tools/transformers/models/llama/convert_to_onnx.py b/onnxruntime/python/tools/transformers/models/llama/convert_to_onnx.py index 7015b8a91b054..d4b06fa096326 100644 --- a/onnxruntime/python/tools/transformers/models/llama/convert_to_onnx.py +++ b/onnxruntime/python/tools/transformers/models/llama/convert_to_onnx.py @@ -1052,8 +1052,7 @@ def main(): logger.info(f"check parity with cmd: {parity_cmd}") parity_check(parity_cmd) except Exception as e: - logger.exception(f"An error occurred while verifying parity: {e}", exc_info=True) - sys.exit(-1) + logger.warning(f"An error occurred while verifying parity: {e}", exc_info=True) if __name__ == "__main__": main()