Fix FA2 tests (#29909)

* fix FA2 tests * refactor inference test name
2024-04-01 08:51:00 +01:00
parent 3b8e2932ce
commit 569f6c7d43
9 changed files with 15 additions and 19 deletions
--- a/tests/models/starcoder2/test_modeling_starcoder2.py
+++ b/tests/models/starcoder2/test_modeling_starcoder2.py
@@ -461,7 +461,7 @@ class Starcoder2ModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTeste
    @require_torch_gpu
    @pytest.mark.flash_attn_test
    @slow
-    def test_flash_attn_2_inference_padding_right(self):
+    def test_flash_attn_2_inference_equivalence_right_padding(self):
        self.skipTest("Starcoder2 flash attention does not support right padding")