Generate: Update VisionEncoderDecoder test value (#27850)
update test result, due to bug fix in decoder-only beam search
This commit is contained in:
@@ -800,7 +800,7 @@ class ViT2GPT2ModelIntegrationTest(unittest.TestCase):
|
|||||||
|
|
||||||
preds, scores = generate_step(pixel_values)
|
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))
|
max_diff = np.amax(np.abs(scores - EXPECTED_SCORES))
|
||||||
self.assertLessEqual(max_diff, 1e-4)
|
self.assertLessEqual(max_diff, 1e-4)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user