"...composable_kernel_onnx.git" did not exist on "b491ebf38480bc0d6cb329ba6825dee610c59097"
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