diff --git a/src/transformers/testing_utils.py b/src/transformers/testing_utils.py index 8e0d7adf6759..b6dfa85b1d20 100644 --- a/src/transformers/testing_utils.py +++ b/src/transformers/testing_utils.py @@ -53,6 +53,7 @@ from .integrations.deepspeed import is_deepspeed_available from .utils import ( ACCELERATE_MIN_VERSION, + GGUF_MIN_VERSION, is_accelerate_available, is_apex_available, is_aqlm_available, @@ -407,7 +408,7 @@ def require_accelerate(test_case, min_version: str = ACCELERATE_MIN_VERSION): )(test_case) -def require_gguf(test_case, min_version: str = "0.10.0"): +def require_gguf(test_case, min_version: str = GGUF_MIN_VERSION): """ Decorator marking a test that requires ggguf. These tests are skipped when gguf isn't installed. """ diff --git a/src/transformers/utils/__init__.py b/src/transformers/utils/__init__.py index 56f594da15f1..6a2ef8437ccf 100755 --- a/src/transformers/utils/__init__.py +++ b/src/transformers/utils/__init__.py @@ -99,6 +99,7 @@ ACCELERATE_MIN_VERSION, ENV_VARS_TRUE_AND_AUTO_VALUES, ENV_VARS_TRUE_VALUES, + GGUF_MIN_VERSION, TORCH_FX_REQUIRED_VERSION, USE_JAX, USE_TF,