From 63d6e0881994cb84ef39a00fa9ff74da9ca3b200 Mon Sep 17 00:00:00 2001 From: kylesayrs Date: Mon, 30 Sep 2024 19:29:31 +0000 Subject: [PATCH 1/2] only check spec --- src/transformers/utils/import_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py index a98b17e4bd57..eb635b3d1ded 100755 --- a/src/transformers/utils/import_utils.py +++ b/src/transformers/utils/import_utils.py @@ -143,7 +143,8 @@ def _is_package_available(pkg_name: str, return_version: bool = False) -> Union[ # `importlib.metadata.version` doesn't work with `awq` _auto_awq_available = importlib.util.find_spec("awq") is not None _quanto_available = _is_package_available("quanto") -_compressed_tensors_available = _is_package_available("compressed_tensors") +# For compressed_tensors, only check spec to allow compress_tensors-nightly package +_compressed_tensors_available = importlib.util.find_spec("compressed_tensors") is not None _pandas_available = _is_package_available("pandas") _peft_available = _is_package_available("peft") _phonemizer_available = _is_package_available("phonemizer") From 11d53a6a11255522a9deb1b371d0a61b6285d642 Mon Sep 17 00:00:00 2001 From: kylesayrs Date: Mon, 30 Sep 2024 20:21:43 +0000 Subject: [PATCH 2/2] correct typo in nightly package name --- src/transformers/utils/import_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py index eb635b3d1ded..2b2302e0dcf7 100755 --- a/src/transformers/utils/import_utils.py +++ b/src/transformers/utils/import_utils.py @@ -143,7 +143,7 @@ def _is_package_available(pkg_name: str, return_version: bool = False) -> Union[ # `importlib.metadata.version` doesn't work with `awq` _auto_awq_available = importlib.util.find_spec("awq") is not None _quanto_available = _is_package_available("quanto") -# For compressed_tensors, only check spec to allow compress_tensors-nightly package +# For compressed_tensors, only check spec to allow compressed_tensors-nightly package _compressed_tensors_available = importlib.util.find_spec("compressed_tensors") is not None _pandas_available = _is_package_available("pandas") _peft_available = _is_package_available("peft")