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

Fix PerceiverModelIntegrationTest::test_inference_masked_lm (#26760)

Browse Source 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-10-12 17:43:06 +02:00
committed by GitHub
parent 33df09e71a
commit a243cdca2a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/perceiver/test_modeling_perceiver.py
View File

@@ -888,7 +888,7 @@ class PerceiverModelIntegrationTest(unittest.TestCase):
logits = outputs.logits logits = outputs.logits
# verify logits # verify logits
expected_shape = torch.Size((1, tokenizer.model_max_length, tokenizer.vocab_size)) expected_shape = torch.Size((1, tokenizer.model_max_length, len(tokenizer)))
self.assertEqual(logits.shape, expected_shape) self.assertEqual(logits.shape, expected_shape)
expected_slice = torch.tensor( expected_slice = torch.tensor(