We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1f4860e commit 123b776Copy full SHA for 123b776
1 file changed
src/transformers/utils/import_utils.py
@@ -38,7 +38,7 @@
38
import packaging.version
39
from packaging import version
40
41
-from . import agnostic, logging
+from . import logging
42
43
44
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
@@ -794,6 +794,7 @@ def is_mamba_2_ssm_available() -> bool:
794
795
@lru_cache
796
def is_flash_linear_attention_available():
797
+ from . import agnostic
798
is_available, fla_version = _is_package_available("fla", return_version=True)
799
return (
800
agnostic.gpu.is_accelerator_available()
0 commit comments