use GGUF_MIN_VERSION everywhere

huggingface · Aug 28, 2024 · bd2b69e · bd2b69e
1 parent 8d66904
commit bd2b69e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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,