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

Fix-ci-whisper (#21767)

Browse Source 
* fix history

* input_features instead of input ids for TFWhisport doctest

* use translate intead of transcribe
This commit is contained in:
Arthur
2023-02-24 11:39:25 +01:00
committed by GitHub
parent c8545d2a9c
commit 087436c98e
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_whisper.py
View File

@@ -1187,7 +1187,7 @@ class WhisperModelIntegrationTests(unittest.TestCase):
input_speech = self._load_datasamples(4)
input_features = processor.feature_extractor(raw_speech=input_speech, return_tensors="pt").input_features
generated_ids = model.generate(input_features, max_length=20)
generated_ids = model.generate(input_features, max_length=20, task="translate")
# fmt: off
EXPECTED_LOGITS = torch.tensor(