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
src/transformers/models/whisper/modeling_tf_whisper.py
View File

@@ -1283,7 +1283,7 @@ class TFWhisperForConditionalGeneration(TFWhisperPreTrainedModel, TFCausalLangua
>>> inputs = processor(ds[0]["audio"]["array"], return_tensors="tf") >>> inputs = processor(ds[0]["audio"]["array"], return_tensors="tf")
>>> input_features = inputs.input_features >>> input_features = inputs.input_features
>>> generated_ids = model.generate(input_ids=input_features) >>> generated_ids = model.generate(input_features=input_features)
>>> transcription = processor.batch_decode(generated_ids, skip_special_tokens=True)[0] >>> transcription = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
>>> transcription >>> transcription

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_speech = self._load_datasamples(4)
input_features = processor.feature_extractor(raw_speech=input_speech, return_tensors="pt").input_features 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 # fmt: off
EXPECTED_LOGITS = torch.tensor( EXPECTED_LOGITS = torch.tensor(