SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[core] Fix fa-2 import (#26785)

Browse Source 
* fix fa-2 import

* nit
This commit is contained in:
Younes Belkada
2023-10-13 12:56:50 +02:00
committed by GitHub
parent 5bfda28dd3
commit 6df9179c1c
7 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/testing_utils.py
View File

@@ -60,7 +60,7 @@ from .utils import (
is_detectron2_available,
is_essentia_available,
is_faiss_available,
is_flash_attn_available,
is_flash_attn_2_available,
is_flax_available,
is_fsdp_available,
is_ftfy_available,
@@ -432,7 +432,7 @@ def require_flash_attn(test_case):
These tests are skipped when Flash Attention isn't installed.
"""
return unittest.skipUnless(is_flash_attn_available(), "test requires Flash Attention")(test_case)
return unittest.skipUnless(is_flash_attn_2_available(), "test requires Flash Attention")(test_case)
def require_peft(test_case):