Skip to content

Commit b81a11a

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

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
@@ -77,7 +77,7 @@ def enforce_single_worker(monkeypatch):
7777
False, "FLASHINFER", False, False, False, False, True, False, False,
7878
False
7979
],
80-
[True, "TRTLLM", False, True, True, True, True, True, True, True, True],
80+
[True, "TRTLLM", False, True, True, True, True, True, True, True],
8181
])
8282
@pytest.mark.high_cuda_memory
8383
def test_llama_eagle3(use_cuda_graph: bool, attn_backend: str,

0 commit comments

Comments
 (0)