From 2d2fa32a41a81116ed97b2091ed74deedc8f9c87 Mon Sep 17 00:00:00 2001 From: Helena Kloosterman Date: Mon, 8 Jan 2024 17:48:51 +0100 Subject: [PATCH] Remove "to ONNX" from info message when exporting model (#1627) --- docs/source/exporters/onnx/usage_guides/export_a_model.mdx | 2 +- optimum/exporters/tasks.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/exporters/onnx/usage_guides/export_a_model.mdx b/docs/source/exporters/onnx/usage_guides/export_a_model.mdx index 1ff74cb11c..4d227e48c2 100644 --- a/docs/source/exporters/onnx/usage_guides/export_a_model.mdx +++ b/docs/source/exporters/onnx/usage_guides/export_a_model.mdx @@ -118,7 +118,7 @@ You should see the following logs (along with potential logs from PyTorch / Tens ```bash Automatic task detection to question-answering. -Framework not specified. Using pt to export to ONNX. +Framework not specified. Using pt to export the model. Using framework PyTorch: 1.12.1 Validating ONNX model... diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index 9973e38743..a5e0522277 100644 --- a/optimum/exporters/tasks.py +++ b/optimum/exporters/tasks.py @@ -1402,7 +1402,7 @@ def determine_framework( else: raise EnvironmentError("Neither PyTorch nor TensorFlow found in environment. Cannot export model.") - logger.info(f"Framework not specified. Using {framework} to export to ONNX.") + logger.info(f"Framework not specified. Using {framework} to export the model.") return framework