Skip to content

GitLab

Unverified Commit 70e4b218 authored by Shangming Cai's avatar Shangming Cai Committed by GitHub
Browse files

Fix flaky logprobs test (#10728) 


Signed-off-by: default avatarShangming Cai <csmthu@gmail.com>
parent 944f1ea0
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/srt/test_logprobs.py
View file @ 70e4b218
...@@ -49,7 +49,7 @@ class TestLogprobsDense(unittest.TestCase): ...@@ -49,7 +49,7 @@ class TestLogprobsDense(unittest.TestCase):
model_path=DENSE_MODEL_NAME, model_path=DENSE_MODEL_NAME,
random_seed=42, random_seed=42,
skip_tokenizer_init=True, skip_tokenizer_init=True,
mem_fraction_static=0.85, mem_fraction_static=0.80,
) )
@classmethod @classmethod
......
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