From 9f5979ee5ce7018301b4109fa72b34264eb4303f Mon Sep 17 00:00:00 2001 From: ZX-ModelCloud Date: Tue, 8 Oct 2024 11:10:48 +0000 Subject: [PATCH] fix assert error --- tests/test_q4_marlin.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_q4_marlin.py b/tests/test_q4_marlin.py index c8e7f8fb0..0f89aa429 100644 --- a/tests/test_q4_marlin.py +++ b/tests/test_q4_marlin.py @@ -176,5 +176,5 @@ def test_outfeatures_288(self): predicted_text = tokenizer.decode(res[0]) - self.assertTrue(predicted_text.startswith(" I am in Paris and I am here to help. I am a very good doctor. I will make you feel better")) + self.assertTrue(predicted_text.startswith(" I am in Paris and I'm going to find you!"))