Update optimum/intel/pipelines/pipeline_base.py

jiqing-feng · echarlaix · web-flow · commit 6d4726bfb61b · 2024-05-06T09:28:16.000+08:00
Co-authored-by: Ella Charlaix &lt;80481427+echarlaix@users.noreply.github.com&gt;
diff --git a/optimum/intel/pipelines/pipeline_base.py b/optimum/intel/pipelines/pipeline_base.py
@@ -240,7 +240,7 @@ def pipeline(
     if accelerator == "ipex":
         if task not in list(IPEX_SUPPORTED_TASKS.keys()):
             raise ValueError(
-                f"Task {task} is not supported for the ONNX Runtime pipeline. Supported tasks are { list(IPEX_SUPPORTED_TASKS.keys())}"
+                f"Task {task} is not supported for the IPEX pipeline. Supported tasks are { list(IPEX_SUPPORTED_TASKS.keys())}"
             )
 
     supported_tasks = IPEX_SUPPORTED_TASKS if accelerator == "ipex" else None

Original file line number	Diff line number	Diff line change
`@@ -240,7 +240,7 @@ def pipeline(`
`240`	`240`	`if accelerator == "ipex":`
`241`	`241`	`if task not in list(IPEX_SUPPORTED_TASKS.keys()):`
`242`	`242`	`raise ValueError(`
`243`		`- f"Task {task} is not supported for the ONNX Runtime pipeline. Supported tasks are { list(IPEX_SUPPORTED_TASKS.keys())}"`
	`243`	`+ f"Task {task} is not supported for the IPEX pipeline. Supported tasks are { list(IPEX_SUPPORTED_TASKS.keys())}"`
`244`	`244`	`)`
`245`	`245`
`246`	`246`	`supported_tasks = IPEX_SUPPORTED_TASKS if accelerator == "ipex" else None`