"...git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "bc7ea032350a5b6f5e4f8c440df7590af5527785"
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
......@@ -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(
......
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