We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 564792f commit b056a2aCopy full SHA for b056a2a
1 file changed
src/transformers/utils/import_utils.py
@@ -795,7 +795,11 @@ def is_mamba_2_ssm_available() -> bool:
795
@lru_cache
796
def is_flash_linear_attention_available():
797
is_available, fla_version = _is_package_available("fla", return_version=True)
798
- return agnostic.gpu.is_accelerator_available() and is_available and version.parse(fla_version) >= version.parse("0.2.2")
+ return (
799
+ agnostic.gpu.is_accelerator_available()
800
+ and is_available
801
+ and version.parse(fla_version) >= version.parse("0.2.2")
802
+ )
803
804
805
0 commit comments