Update optimum/intel/pipelines/pipeline_base.py

jiqing-feng · echarlaix · web-flow · commit 7f92191f6e4c · 2024-05-15T14:05:11.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
@@ -30,10 +30,8 @@
 )
 from transformers.pipelines.base import Pipeline
 from transformers.tokenization_utils import PreTrainedTokenizer
-from transformers.utils import (
-    is_ipex_available,
-    logging,
-)
+from transformers.utils import logging
+from optimum.intel.utils import is_ipex_available
 
 
 if is_ipex_available():
