From add0895dd98c1a193bf0ada2f3575fcd5e256cb2 Mon Sep 17 00:00:00 2001 From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Date: Fri, 28 Jul 2023 16:45:09 +0200 Subject: [PATCH] [`Mpt`] Fix mpt slow test (#25170) fix mpt slow test --- tests/models/mpt/test_modeling_mpt.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/models/mpt/test_modeling_mpt.py b/tests/models/mpt/test_modeling_mpt.py index f3fc6d35951..cb39fd08f52 100644 --- a/tests/models/mpt/test_modeling_mpt.py +++ b/tests/models/mpt/test_modeling_mpt.py @@ -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)