riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Fix is_torch_xpu_available for torch < 2.3 (#31573)

Browse Source
This commit is contained in:
amyeroberts 2024-06-24 16:57:49 +01:00 committed by GitHub
parent 2fc9d8e9b1
commit 3a49ebe0d8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/transformers/utils/import_utils.py
View File

@ -754,10 +754,13 @@ def is_torch_xpu_available(check_device=False):
if not is_torch_available():
return False
import torch
torch_version = version.parse(_torch_version)
if is_ipex_available():
import intel_extension_for_pytorch # noqa: F401
elif torch_version.major < 2 or (torch_version.major == 2 and torch_version.minor < 4):
return False
import torch
if check_device:
try: