Skip to content

GitLab

Unverified Commit 761b3fad authored by Arthur's avatar Arthur Committed by GitHub
Browse files

Expected output for the test changed (#20493)

parent a4beb37b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/whisper/test_modeling_tf_whisper.py
View file @ 761b3fad
...@@ -743,7 +743,7 @@ def _test_large_generation_multilingual(in_queue, out_queue, timeout): ...@@ -743,7 +743,7 @@ def _test_large_generation_multilingual(in_queue, out_queue, timeout):
) )
transcript = processor.batch_decode(generated_ids, skip_special_tokens=True)[0] transcript = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
EXPECTED_TRANSCRIPT = " Kimura san ni denwa wo kaite moraimashita" EXPECTED_TRANSCRIPT = " Kimura-san called me."
unittest.TestCase().assertEqual(transcript, EXPECTED_TRANSCRIPT) unittest.TestCase().assertEqual(transcript, EXPECTED_TRANSCRIPT)
model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(language="ja", task="translate") model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(language="ja", task="translate")
......
tests/models/whisper/test_modeling_whisper.py
View file @ 761b3fad
...@@ -983,7 +983,7 @@ class WhisperModelIntegrationTests(unittest.TestCase): ...@@ -983,7 +983,7 @@ class WhisperModelIntegrationTests(unittest.TestCase):
) )
transcript = processor.batch_decode(generated_ids, skip_special_tokens=True)[0] transcript = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
EXPECTED_TRANSCRIPT = " Kimura san ni denwa wo kaite moraimashita" EXPECTED_TRANSCRIPT = " Kimura-san called me."
self.assertEqual(transcript, EXPECTED_TRANSCRIPT) self.assertEqual(transcript, EXPECTED_TRANSCRIPT)
model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(language="ja", task="translate") model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(language="ja", task="translate")
......
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