Fix flaky logprobs test (#10728)
Signed-off-by: Shangming Cai <csmthu@gmail.com>
This commit is contained in:
@@ -49,7 +49,7 @@ class TestLogprobsDense(unittest.TestCase):
|
||||
model_path=DENSE_MODEL_NAME,
|
||||
random_seed=42,
|
||||
skip_tokenizer_init=True,
|
||||
mem_fraction_static=0.85,
|
||||
mem_fraction_static=0.80,
|
||||
)
|
||||
|
||||
@classmethod
|
||||
|
||||
Reference in New Issue
Block a user