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

[Mpt] Fix mpt slow test (#25170)

Browse Source 
fix mpt slow test
This commit is contained in:
Younes Belkada
2023-07-28 16:45:09 +02:00
committed by GitHub
parent d53b8ad780
commit add0895dd9
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/mpt/test_modeling_mpt.py
View File

@@ -432,7 +432,7 @@ class MptIntegrationTests(unittest.TestCase):
) )
input_text = "Hello" input_text = "Hello"
expected_output = "Hello my name is [name] and I am a [type] at [company]. I have a [number]" expected_output = """Hello, I\'m a new user of the forum. I have a question about the "Solaris"""
inputs = tokenizer(input_text, return_tensors="pt") inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=20) outputs = model.generate(**inputs, max_new_tokens=20)
@@ -450,7 +450,9 @@ class MptIntegrationTests(unittest.TestCase):
) )
input_text = "Hello" input_text = "Hello"
expected_output = "Hello my name is Kaitlyn and I am a senior at the University of Wisconsin-Stout. I am major" expected_output = (
"Hello and welcome to the first day of the new release countdown for the month of May!\nToday"
)
inputs = tokenizer(input_text, return_tensors="pt") inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=20) outputs = model.generate(**inputs, max_new_tokens=20)