Skip to content

GitLab

Unverified Commit bf82c9b7 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`BLIP`] Fix daily CI failing test (#20877)

parent beb24f2a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/models/blip/test_modeling_blip.py
View file @ bf82c9b7
......@@ -853,7 +853,7 @@ class BlipModelIntegrationTest(unittest.TestCase):
out_itm = model(**inputs)
out = model(**inputs, use_itm_head=False)
expected_scores = torch.Tensor([[0.9779, 0.0221]])
expected_scores = torch.Tensor([[0.9798, 0.0202]])
self.assertTrue(torch.allclose(torch.nn.Softmax()(out_itm[0].cpu()), expected_scores, atol=1e-3, rtol=1e-3))
self.assertTrue(torch.allclose(out[0].cpu(), torch.Tensor([[0.5053]]), atol=1e-3, rtol=1e-3))
self.assertTrue(torch.allclose(torch.nn.Softmax()(out_itm[0].cpu()), expected_scores, rtol=1e-3, atol=1e-3))
self.assertTrue(torch.allclose(out[0].cpu(), torch.Tensor([[0.5053]]), rtol=1e-3, atol=1e-3))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment