We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6d25706 commit 0ff77abCopy full SHA for 0ff77ab
1 file changed
src/transformers/utils/import_utils.py
@@ -795,6 +795,7 @@ def is_mamba_2_ssm_available() -> bool:
795
@lru_cache
796
def is_flash_linear_attention_available():
797
from . import agnostic
798
+
799
is_available, fla_version = _is_package_available("fla", return_version=True)
800
return (
801
agnostic.gpu.is_accelerator_available()
0 commit comments