From d231f5a7d4d110fffe91ba31b4995c8574036c14 Mon Sep 17 00:00:00 2001 From: jiqing-feng Date: Fri, 9 May 2025 04:35:24 +0800 Subject: [PATCH] update bnb tests (#38011) Signed-off-by: jiqing-feng --- tests/quantization/bnb/test_4bit.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/quantization/bnb/test_4bit.py b/tests/quantization/bnb/test_4bit.py index 967e2cfa16..5c7ef7a915 100644 --- a/tests/quantization/bnb/test_4bit.py +++ b/tests/quantization/bnb/test_4bit.py @@ -109,6 +109,7 @@ class Base4bitTest(unittest.TestCase): EXPECTED_OUTPUTS.add("Hello my name is John.\nI am a friend of your father.\n") EXPECTED_OUTPUTS.add("Hello my name is John Doe, I am a student at the University") EXPECTED_OUTPUTS.add("Hello my name is John and I am 25 years old.") + EXPECTED_OUTPUTS.add("Hello my name is John and I am a student at the University of") MAX_NEW_TOKENS = 10 def setUp(self):