FIX / TST: Fix expected results on Mistral slow test (A10) (#30909)

Update test_modeling_mistral.py
2024-05-21 09:14:14 +02:00
parent 0df888ffb7
commit c876d12127
1 changed files with 1 additions and 1 deletions
--- 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)