Skip to content

Commit

Permalink
TST Bump absolute tolerance for test (#1891)
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
kallewoof committed Jul 2, 2024
1 parent 62122b5 commit 1e5227f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_common_gpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 1e5227f

Please sign in to comment.