"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "803a8cd18f44e7288187b82ae3b48956edf11dd6"
Unverified Commit add0895d authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`Mpt`] Fix mpt slow test (#25170)

fix mpt slow test
parent d53b8ad7
...@@ -432,7 +432,7 @@ class MptIntegrationTests(unittest.TestCase): ...@@ -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): ...@@ -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)
......
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