fix is_intel_gpu_available (#1527)

intel_extension_for_transformers/tools/utils.py

@@ -38,7 +38,6 @@ def is_intel_gpu_available():
     import intel_extension_for_pytorch as ipex
     return hasattr(torch, "xpu") and torch.xpu.is_available()
 
-
 _ipex_available = importlib.util.find_spec("intel_extension_for_pytorch") is not None
 _ipex_version = "N/A"
 if _ipex_available:

requirements.txt

@@ -1,5 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/cpu
 py-cpuinfo
 setuptools>=65
 setuptools_scm[toml]>=6.2
-torch==2.2.0

setup.py

@@ -11,7 +11,11 @@
 
 result = subprocess.Popen("pip install -r requirements.txt", shell=True)
 result.wait()
-from intel_extension_for_transformers.tools.utils import is_intel_gpu_available
+
+def is_intel_gpu_available():
+    import torch
+    import intel_extension_for_pytorch as ipex
+    return hasattr(torch, "xpu") and torch.xpu.is_available()
 
 def check_env_flag(name: str, default: bool = False) -> bool:
     if default:  # if a flag meant to be true if not set / mal-formatted