Generate: Update VisionEncoderDecoder test value (#27850)

update test result, due to bug fix in decoder-only beam search

Generate: Update VisionEncoderDecoder test value (#27850)
update test result, due to bug fix in decoder-only beam search
b7e6d120 · Joao Gante · GitHub · fdb85be4 · b7e6d120
Unverified Commit b7e6d120 authored Dec 05, 2023 by Joao Gante Committed by GitHub Dec 05, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py ...n_encoder_decoder/test_modeling_vision_encoder_decoder.py +1 -1

No files found.
--- 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)