SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
Update test_modeling_mistral.py
This commit is contained in:
Younes Belkada
2024-05-21 09:14:14 +02:00
committed by GitHub
parent 0df888ffb7
commit c876d12127
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/mistral/test_modeling_mistral.py
View File

@@ -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)