riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats 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"
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")
outputs = model.generate(**inputs, max_new_tokens=20)
@ -450,7 +450,9 @@ class MptIntegrationTests(unittest.TestCase):
)
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")
outputs = model.generate(**inputs, max_new_tokens=20)