From 11b670a2824accf1b22b33694ab42e2f17ad8af2 Mon Sep 17 00:00:00 2001
From: Yuanyuan Chen <cyyever@outlook.com>
Date: Fri, 23 May 2025 16:18:30 +0800
Subject: [PATCH] Fix run_slow (#38314)

Signed-off-by: cyy <cyyever@outlook.com>
---
 tests/generation/test_paged_attention.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/generation/test_paged_attention.py b/tests/generation/test_paged_attention.py
index c06975844bf..566a4ed0078 100644
--- a/tests/generation/test_paged_attention.py
+++ b/tests/generation/test_paged_attention.py
@@ -4,7 +4,7 @@ import unittest
 from parameterized import parameterized
 
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
-from transformers.testing_utils import require_flash_attn, require_torch_gpu, run_slow
+from transformers.testing_utils import require_flash_attn, require_torch_gpu, slow
 
 
 _TEST_PROMPTS = [
@@ -24,7 +24,7 @@ _EXPECTED_OUTPUTS = [
 ]
 
 
-@run_slow
+@slow
 @require_torch_gpu
 @require_flash_attn
 class TestBatchGeneration(unittest.TestCase):