Remove "to ONNX" from info message when exporting model (#1627)

huggingface · Jan 8, 2024 · 0f0a663 · 0f0a663
1 parent ec14b3f
commit 0f0a663
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 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
@@ -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
@@ -1403,7 +1403,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