[ci] Update rtol for test_classification (#36556)
Signed-off-by: angelayi <yiangela7@gmail.com> Co-authored-by: Richard Zou <zou3519@users.noreply.github.com>
This commit is contained in:
@@ -45,5 +45,7 @@ def test_models(
|
|||||||
# half datatype tests in
|
# half datatype tests in
|
||||||
# tests/models/language/pooling/test_embedding.py
|
# tests/models/language/pooling/test_embedding.py
|
||||||
assert torch.allclose(
|
assert torch.allclose(
|
||||||
hf_output, vllm_output, 1e-3 if dtype == "float" else 1e-2
|
hf_output,
|
||||||
|
vllm_output,
|
||||||
|
rtol=2e-3 if dtype == "float" else 1e-2,
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user