From b7e6d120c1c22384df285b06be9e63a81ddfe5c0 Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Tue, 5 Dec 2023 11:26:59 +0000 Subject: [PATCH] Generate: Update VisionEncoderDecoder test value (#27850) update test result, due to bug fix in decoder-only beam search --- .../test_modeling_vision_encoder_decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py b/tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py index 8ac4449cf3..40862c6234 100644 --- 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)