riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Correct @is_flaky test decoration (#31480)

Browse Source 
* Correct @is_flaky decorator
This commit is contained in:
Pavel Iakubovskii 2024-06-24 08:09:21 +01:00 committed by GitHub
parent 4b822560a1
commit 3c2d4d60d7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/gemma/test_modeling_gemma.py
View File

@ -493,7 +493,7 @@ class GemmaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
@require_flash_attn
@require_torch_gpu
@pytest.mark.flash_attn_test
@is_flaky
@is_flaky()
@slow
def test_flash_attn_2_equivalence(self):
for model_class in self.all_model_classes:

8
tests/test_modeling_common.py
View File

@ -3407,7 +3407,7 @@ class ModelTesterMixin:
@require_torch_gpu
@mark.flash_attn_test
@slow
@is_flaky
@is_flaky()
def test_flash_attn_2_inference_equivalence(self):
for model_class in self.all_model_classes:
if not model_class._supports_flash_attn_2:
@ -3501,7 +3501,7 @@ class ModelTesterMixin:
@require_torch_gpu
@mark.flash_attn_test
@slow
@is_flaky
@is_flaky()
def test_flash_attn_2_inference_equivalence_right_padding(self):
for model_class in self.all_model_classes:
if not model_class._supports_flash_attn_2:
@ -3591,7 +3591,7 @@ class ModelTesterMixin:
@require_torch_gpu
@mark.flash_attn_test
@slow
@is_flaky
@is_flaky()
def test_flash_attn_2_generate_left_padding(self):
for model_class in self.all_generative_model_classes:
if not model_class._supports_flash_attn_2:
@ -3635,7 +3635,7 @@ class ModelTesterMixin:
@require_flash_attn
@require_torch_gpu
@mark.flash_attn_test
@is_flaky
@is_flaky()
@slow
def test_flash_attn_2_generate_padding_right(self):
for model_class in self.all_generative_model_classes: