Generate: Update VisionEncoderDecoder test value (#27850)

update test result, due to bug fix in decoder-only beam search
2023-12-05 11:26:59 +00:00
parent fdb85be40f
commit b7e6d120c1
1 changed files with 1 additions and 1 deletions
--- a/tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py
+++ b/tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py
@@ -800,7 +800,7 @@ class ViT2GPT2ModelIntegrationTest(unittest.TestCase):

        preds, scores = generate_step(pixel_values)

-        EXPECTED_SCORES = np.array([-0.59562886])
+        EXPECTED_SCORES = np.array([-0.64145195])
        max_diff = np.amax(np.abs(scores - EXPECTED_SCORES))
        self.assertLessEqual(max_diff, 1e-4)