riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 12:50:06 +06:00
Code Issues Actions 28 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats 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):