From 62fa18ea73e6b67fd7b98fdc8098a614fd6abdf8 Mon Sep 17 00:00:00 2001 From: Yuanyuan Chen Date: Wed, 17 Sep 2025 11:48:05 +0800 Subject: [PATCH] Update torch minimum version check to 2.2 Signed-off-by: Yuanyuan Chen --- src/transformers/utils/import_utils.py | 4 ++-- tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py | 2 -- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py index 2f6dc0b8e714..3eb1b60e328d 100644 --- a/src/transformers/utils/import_utils.py +++ b/src/transformers/utils/import_utils.py @@ -248,9 +248,9 @@ def _is_package_available(pkg_name: str, return_version: bool = False) -> Union[ if USE_TORCH in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TF not in ENV_VARS_TRUE_VALUES: _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}") else: logger.info("Disabling PyTorch because USE_TF is set") _torch_available = False 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 a105302a9952..10c1470af517 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