anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[CI] Fix test. (#2031)

Browse Source 
Signed-off-by: Alicia <115451386+congw729@users.noreply.github.com>
This commit is contained in:
Alicia authored and GitHub committed 2026-03-20 16:41:45 +08:00
1 parent fafa248665
commit 199f7832e7
3 files changed
+18 -17
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/model_executor/models/voxtral_tts/test_cuda_graph_acoustic_transformer.py
+1
View File
@@ -23,6 +23,7 @@ pytestmark = [
pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA required"),
pytest.mark.core_model,
pytest.mark.cuda,
pytest.mark.L4,
]
DEVICE = torch.device("cuda:0")