Disable message about ONNX configs on export (#659)

helena-intel · web-flow · commit 402b9dba1284 · 2024-04-12T09:37:12.000+02:00
diff --git a/optimum/exporters/openvino/convert.py b/optimum/exporters/openvino/convert.py
@@ -564,6 +564,7 @@ def export_from_model(
             kwargs_shapes[input_name] if input_name in kwargs_shapes else DEFAULT_DUMMY_SHAPES[input_name]
         )
 
+    logging.disable(logging.INFO)
     export_config, models_and_export_configs = _get_submodels_and_export_configs(
         model=model,
         task=task,
@@ -578,6 +579,7 @@ def export_from_model(
         legacy=False,
         exporter="openvino",
     )
+    logging.disable(logging.NOTSET)
 
     if ov_config is None:
         if library_name == "diffusers":

Original file line number	Diff line number	Diff line change
`@@ -564,6 +564,7 @@ def export_from_model(`
`564`	`564`	`kwargs_shapes[input_name] if input_name in kwargs_shapes else DEFAULT_DUMMY_SHAPES[input_name]`
`565`	`565`	`)`
`566`	`566`
	`567`	`+ logging.disable(logging.INFO)`
`567`	`568`	`export_config, models_and_export_configs = _get_submodels_and_export_configs(`
`568`	`569`	`model=model,`
`569`	`570`	`task=task,`
`@@ -578,6 +579,7 @@ def export_from_model(`
`578`	`579`	`legacy=False,`
`579`	`580`	`exporter="openvino",`
`580`	`581`	`)`
	`582`	`+ logging.disable(logging.NOTSET)`
`581`	`583`
`582`	`584`	`if ov_config is None:`
`583`	`585`	`if library_name == "diffusers":`