riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 12:50:06 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

Remove IPEX requirement for bitsandbytes on CPU (#38594)

Browse Source 
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
This commit is contained in:
Matthew Douglas 2025-06-11 11:46:34 -04:00 committed by GitHub
parent 063bef0865
commit e2bdc13375
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/transformers/integrations/bitsandbytes.py
View File

@ -11,7 +11,6 @@ from ..utils import (
is_accelerate_available, is_accelerate_available,
is_bitsandbytes_available, is_bitsandbytes_available,
is_bitsandbytes_multi_backend_available, is_bitsandbytes_multi_backend_available,
is_ipex_available,
is_torch_available, is_torch_available,
logging, logging,
) )
@ -488,29 +487,11 @@ def _validate_bnb_multi_backend_availability(raise_exception):
bnb_supported_devices = getattr(bnb, "supported_torch_devices", set()) bnb_supported_devices = getattr(bnb, "supported_torch_devices", set())
available_devices = set(get_available_devices()) available_devices = set(get_available_devices())
if available_devices == {"cpu"} and not is_ipex_available():
from importlib.util import find_spec
if find_spec("intel_extension_for_pytorch"):
logger.warning(
"You have Intel IPEX installed but if you're intending to use it for CPU, it might not have the right version. Be sure to double check that your PyTorch and IPEX installs are compatible."
)
available_devices = frozenset(
[device for device in available_devices if device != "cpu"]
) # Only Intel CPU is supported by BNB at the moment
if not available_devices.intersection(bnb_supported_devices): if not available_devices.intersection(bnb_supported_devices):
if raise_exception: if raise_exception:
bnb_supported_devices_with_info = set( # noqa: C401
'"cpu" (needs an Intel CPU and intel_extension_for_pytorch installed and compatible with the PyTorch version)'
if device == "cpu"
else device
for device in bnb_supported_devices
)
err_msg = ( err_msg = (
f"None of the available devices `available_devices = {available_devices or None}` are supported by the bitsandbytes version you have installed: `bnb_supported_devices = {bnb_supported_devices_with_info}`. " f"None of the available devices `available_devices = {available_devices or None}` are supported by the bitsandbytes version you have installed: `bnb_supported_devices = {bnb_supported_devices}`. "
"Please check the docs to see if the backend you intend to use is available and how to install it: https://huggingface.co/docs/bitsandbytes/main/en/installation#multi-backend" "Please check the docs to see if the backend you intend to use is available and how to install it: https://huggingface.co/docs/bitsandbytes/main/en/installation"
) )
logger.error(err_msg) logger.error(err_msg)