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

Fix run_slow (#38314)

Browse Source 
Signed-off-by: cyy <cyyever@outlook.com>
This commit is contained in:
Yuanyuan Chen
2025-05-23 16:18:30 +08:00
committed by GitHub
parent b01984a51d
commit 11b670a282
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/generation/test_paged_attention.py
View File

@@ -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):