use pytest.mark directly (#27390)

fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

tests/models/llama/test_modeling_llama.py

@@ -17,8 +17,8 @@
 
 import unittest
 
+import pytest
 from parameterized import parameterized
-from pytest import mark
 
 from transformers import LlamaConfig, is_torch_available, set_seed
 from transformers.testing_utils import (
@@ -385,7 +385,7 @@ class LlamaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
 
     @require_flash_attn
     @require_torch_gpu
-    @mark.flash_attn_test
+    @pytest.mark.flash_attn_test
     @slow
     def test_flash_attn_2_generate_padding_right(self):
         """