From a243cdca2a593ec5c547d67601d645e569963712 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 12 Oct 2023 17:43:06 +0200 Subject: [PATCH] Fix `PerceiverModelIntegrationTest::test_inference_masked_lm` (#26760) fix Co-authored-by: ydshieh --- tests/models/perceiver/test_modeling_perceiver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/perceiver/test_modeling_perceiver.py b/tests/models/perceiver/test_modeling_perceiver.py index 91fac90e7b..aeb9b80deb 100644 --- a/tests/models/perceiver/test_modeling_perceiver.py +++ b/tests/models/perceiver/test_modeling_perceiver.py @@ -888,7 +888,7 @@ class PerceiverModelIntegrationTest(unittest.TestCase): logits = outputs.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) expected_slice = torch.tensor(