kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix

Browse Source 
Signed-off-by: Xiwen Yu <13230610+VALLIS-NERIA@users.noreply.github.com>
This commit is contained in:
Xiwen Yu 2025-09-15 15:18:55 +08:00
parent d48e82a21f
commit 7657d83553
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/unittest/llmapi/test_llm.py
View File

@ -1882,7 +1882,6 @@ def llm_return_logprobs_test_harness(prompt_logprobs: Optional[int],
@pytest.mark.parametrize(
"prompt_logprobs, logprobs, return_context_logits, return_generation_logits",
[(2, None, True, False), (None, 2, False, False)])
@pytest.skip(reason="https://nvbugspro.nvidia.com/bug/5516849")
def test_llm_return_logprobs(prompt_logprobs: Optional[int],
logprobs: Optional[int],
return_context_logits: bool,
@ -1894,7 +1893,6 @@ def test_llm_return_logprobs(prompt_logprobs: Optional[int],
@pytest.mark.skip(reason="https://nvbugs/5516660")
@force_ampere
@pytest.skip(reason="https://nvbugspro.nvidia.com/bug/5516849")
def test_llm_return_logprobs_streaming():
llm_return_logprobs_test_harness(2, 2, False, True, streaming=True)