Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "cd09a8dfbce5f4696ba5ec0c711567df214571c0"
Unverified Commit a243cdca authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `PerceiverModelIntegrationTest::test_inference_masked_lm` (#26760) 



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 33df09e7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/perceiver/test_modeling_perceiver.py
View file @ a243cdca
...@@ -888,7 +888,7 @@ class PerceiverModelIntegrationTest(unittest.TestCase): ...@@ -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(
......
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