From c876d12127272ce2886ce51399928c3343d475b9 Mon Sep 17 00:00:00 2001 From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Date: Tue, 21 May 2024 09:14:14 +0200 Subject: [PATCH] FIX / TST: Fix expected results on Mistral slow test (A10) (#30909) Update test_modeling_mistral.py --- tests/models/mistral/test_modeling_mistral.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/mistral/test_modeling_mistral.py b/tests/models/mistral/test_modeling_mistral.py index 13bdf83c5b..dfa0646f8f 100644 --- a/tests/models/mistral/test_modeling_mistral.py +++ b/tests/models/mistral/test_modeling_mistral.py @@ -627,7 +627,7 @@ class MistralIntegrationTest(unittest.TestCase): def test_speculative_generation(self): EXPECTED_TEXT_COMPLETION = { 7: "My favourite condiment is 100% Sriracha. I love the heat, the tang and the fact costs", - 8: "My favourite condiment is 100% Sriracha. I love the heat, the sweetness, the tang", + 8: "My favourite condiment is 100% ketchup. I love it on everything. I’m not a big", } prompt = "My favourite condiment is " tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", use_fast=False)