From 1e5227ff90b3501485544de1b7407a87c5cf9d17 Mon Sep 17 00:00:00 2001 From: kallewoof Date: Tue, 2 Jul 2024 18:37:43 +0900 Subject: [PATCH] TST Bump absolute tolerance for test (#1891) The test test_4bit_lora_mixed_adapter_batches_lora allclose can fail on some systems, even though it passes on others (like CI). Increase the tolerance slightly to get rid of this. --- tests/test_common_gpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_common_gpu.py b/tests/test_common_gpu.py index cd9d3cd943..b490bd7de9 100644 --- a/tests/test_common_gpu.py +++ b/tests/test_common_gpu.py @@ -801,7 +801,7 @@ def test_4bit_lora_mixed_adapter_batches_lora(self): with torch.inference_mode(): out_adapter1 = model(**inputs).logits - atol, rtol = 1e-5, 1e-5 + atol, rtol = 3e-5, 1e-5 # sanity check, outputs have the right shape and are not the same assert len(out_base) >= 3 assert len(out_base) == len(out_adapter0) == len(out_adapter1)