From 8ceeb016b6e84f636053be5a1af809dc32eed99f Mon Sep 17 00:00:00 2001 From: SeongBeomLEE <2712qwer@naver.com> Date: Thu, 11 Jul 2024 20:45:25 +0900 Subject: [PATCH] [fix] AttributeError in is_flash_attn_greater_or_equal Changed the local variable version to pkg_version. --- src/transformers/utils/import_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py index d8a4cdc61c83..af272c43a930 100755 --- a/src/transformers/utils/import_utils.py +++ b/src/transformers/utils/import_utils.py @@ -812,11 +812,11 @@ def is_flash_attn_greater_or_equal_2_10(): return version.parse(importlib.metadata.version("flash_attn")) >= version.parse("2.1.0") -def is_flash_attn_greater_or_equal(version): +def is_flash_attn_greater_or_equal(pkg_version): if not _is_package_available("flash_attn"): return False - return version.parse(importlib.metadata.version("flash_attn")) >= version.parse(version) + return version.parse(importlib.metadata.version("flash_attn")) >= version.parse(pkg_version) def is_torchdistx_available():