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,
|
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
|
||||||
|
|||||||
Reference in New Issue
Block a user