From 0ddda212bdd6bb9a56bf05276d82dcf23a2c9a86 Mon Sep 17 00:00:00 2001 From: yonigozlan Date: Fri, 22 Nov 2024 23:12:29 +0000 Subject: [PATCH] fix slow test --- tests/models/got_ocr2/test_modeling_got_ocr2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/got_ocr2/test_modeling_got_ocr2.py b/tests/models/got_ocr2/test_modeling_got_ocr2.py index 1f53b95bbdbc31..6e0f86d7b67fae 100644 --- a/tests/models/got_ocr2/test_modeling_got_ocr2.py +++ b/tests/models/got_ocr2/test_modeling_got_ocr2.py @@ -278,7 +278,7 @@ def test_small_model_integration_test_got_ocr_format(self): decoded_output = self.processor.decode( generate_ids[0, inputs["input_ids"].shape[1] :], skip_special_tokens=True ) - expected_output = "\\title{\nR'" + expected_output = "\\title{\nR" self.assertEqual(decoded_output, expected_output) @slow