SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Younes Belkada
2023-01-05 13:24:31 +01:00
committed by GitHub
parent beb24f2a36
commit bf82c9b74f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/blip/test_modeling_blip.py
View File

@@ -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))