Skip to content

Commit a7276c8

Browse files
committed
fix test
Signed-off-by: Dylan Chen <[email protected]>
1 parent 2436739 commit a7276c8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unittest/_torch/speculative/test_eagle3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def enforce_single_worker(monkeypatch):
7878
False, "FLASHINFER", False, False, False, False, True, False, False,
7979
False
8080
],
81-
[True, "TRTLLM", False, True, True, True, True, True, True, True, True],
81+
[True, "TRTLLM", False, True, True, True, True, True, True, True],
8282
])
8383
@pytest.mark.high_cuda_memory
8484
def test_llama_eagle3(use_cuda_graph: bool, attn_backend: str,

0 commit comments

Comments
 (0)