Unverified Commit c876d121 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

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

Update test_modeling_mistral.py
parent 0df888ff
...@@ -627,7 +627,7 @@ class MistralIntegrationTest(unittest.TestCase): ...@@ -627,7 +627,7 @@ class MistralIntegrationTest(unittest.TestCase):
def test_speculative_generation(self): def test_speculative_generation(self):
EXPECTED_TEXT_COMPLETION = { EXPECTED_TEXT_COMPLETION = {
7: "My favourite condiment is 100% Sriracha. I love the heat, the tang and the fact costs", 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 " prompt = "My favourite condiment is "
tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", use_fast=False) tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", use_fast=False)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment