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

Fix tests (#32649)

Browse Source 
* skip failing tests

* [no-filter]

* [no-filter]

* fix wording catch in FA2 test

* [no-filter]

* trigger normal CI without filtering
This commit is contained in:
Pablo Montalvo
2024-08-13 10:46:21 +02:00
committed by GitHub
parent 29c3a0fa01
commit a5a8291ad1
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/utils/test_modeling_utils.py
View File

@@ -2427,8 +2427,7 @@ class TestAttentionImplementation(unittest.TestCase):
_ = AutoModel.from_pretrained(
"hf-internal-testing/tiny-random-GPTBigCodeModel", attn_implementation="flash_attention_2"
)
self.assertTrue("the package flash_attn seems not to be installed" in str(cm.exception))
self.assertTrue("the package flash_attn seems to be not installed" in str(cm.exception))
def test_not_available_flash_with_config(self):
if is_flash_attn_2_available():
@@ -2443,7 +2442,7 @@ class TestAttentionImplementation(unittest.TestCase):
attn_implementation="flash_attention_2",
)
self.assertTrue("the package flash_attn seems not to be installed" in str(cm.exception))
self.assertTrue("the package flash_attn seems to be not installed" in str(cm.exception))
def test_not_available_sdpa(self):
if is_torch_sdpa_available():