SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Expected output for the test changed (#20493)

Browse Source
This commit is contained in:
Arthur
2022-11-30 15:07:28 +01:00
committed by GitHub
parent a4beb37b81
commit 761b3fad92
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/whisper/test_modeling_tf_whisper.py
View File

@@ -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")

2
tests/models/whisper/test_modeling_whisper.py
View File

@@ -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")