diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py index df06bd05842d..76cbaad0eb17 100644 --- a/src/transformers/utils/import_utils.py +++ b/src/transformers/utils/import_utils.py @@ -236,9 +236,9 @@ def _is_package_available(pkg_name: str, return_version: bool = False) -> Union[ _torch_available, _torch_version = _is_package_available("torch", return_version=True) if _torch_available: - _torch_available = version.parse(_torch_version) >= version.parse("2.1.0") + _torch_available = version.parse(_torch_version) >= version.parse("2.2.0") if not _torch_available: - logger.warning(f"Disabling PyTorch because PyTorch >= 2.1 is required but found {_torch_version}") + logger.warning(f"Disabling PyTorch because PyTorch >= 2.2 is required but found {_torch_version}") _essentia_available = importlib.util.find_spec("essentia") is not None diff --git a/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py b/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py index d90dff9f13ff..cb2e31867194 100644 --- a/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py +++ b/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py @@ -55,8 +55,6 @@ if is_torch_available(): import torch -else: - is_torch_greater_or_equal_than_2_0 = False if is_vision_available(): from PIL import Image