From fbf4c6a58b8750578f9f90686504d3505783487a Mon Sep 17 00:00:00 2001 From: Marc Sun Date: Wed, 22 May 2024 16:40:54 +0200 Subject: [PATCH] fix awq mistral test --- tests/quantization/autoawq/test_awq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/quantization/autoawq/test_awq.py b/tests/quantization/autoawq/test_awq.py index 20ecd783cf04..aaab7226c870 100644 --- a/tests/quantization/autoawq/test_awq.py +++ b/tests/quantization/autoawq/test_awq.py @@ -288,7 +288,7 @@ class AwqFusedTest(unittest.TestCase): "You end up exactly where you started. Where are you?" ) - EXPECTED_GENERATION = prompt + "\n\nYou are at the starting point.\n\nIf" + EXPECTED_GENERATION = prompt + "\n\nThis is a classic puzzle that has been around for" EXPECTED_GENERATION_CUSTOM_MODEL = "Hello,\n\nI have a problem with my 20" EXPECTED_GENERATION_MIXTRAL = prompt + " You're on the North Pole.\n\nThe"