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

[X-CLIP] Fix doc tests (#19523)

Browse Source 
* Fix XCLIP doc tests

* Add model to doc test list

* Fix tests
This commit is contained in:
NielsRogge
2022-10-12 17:05:12 +02:00
committed by GitHub
parent eefcecaa35
commit 9e29080439
4 changed files with 111 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/x_clip/test_modeling_x_clip.py
View File

@@ -667,6 +667,6 @@ class XCLIPModelIntegrationTest(unittest.TestCase):
torch.Size((inputs.input_ids.shape[0], inputs.pixel_values.shape[0])),
)
expected_logits = torch.tensor([[14.3819, 20.6031, 15.0526]], device=torch_device)
expected_logits = torch.tensor([[14.0181, 20.2771, 14.4776]], device=torch_device)
self.assertTrue(torch.allclose(outputs.logits_per_video, expected_logits, atol=1e-3))