Skip to content

GitLab

Unverified Commit 20453cef authored by Byron Hsu's avatar Byron Hsu Committed by GitHub
Browse files

[test] Lower number of top logprobs to get rid of `-inf` (#3212)

parent 9f635ea5
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/srt/sampling/penaltylib/test_srt_endpoint_with_penalizers.py
View file @ 20453cef
...@@ -36,7 +36,7 @@ class TestBatchPenalizerE2E(unittest.TestCase): ...@@ -36,7 +36,7 @@ class TestBatchPenalizerE2E(unittest.TestCase):
def run_decode( def run_decode(
self, self,
return_logprob=True, return_logprob=True,
top_logprobs_num=5, top_logprobs_num=3,
return_text=True, return_text=True,
n=1, n=1,
**sampling_params, **sampling_params,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment