riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-30 17:52:35 +06:00
Code Issues Actions 30 Packages Projects Releases Wiki Activity

add missing alignment_heads to Whisper integration test (#24487)

Browse Source 
add missing alignment heads
This commit is contained in:
Matthijs Hollemans 2023-06-26 11:50:10 +02:00 committed by GitHub
parent 868363abb9
commit 3b84d86b57
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1442,6 +1442,7 @@ class WhisperModelIntegrationTests(unittest.TestCase):
processor = WhisperProcessor.from_pretrained("openai/whisper-tiny")
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny")
model.to(torch_device)
model.generation_config.alignment_heads = [[2, 2], [3, 0], [3, 2], [3, 3], [3, 4], [3, 5]]
input_speech = self._load_datasamples(4)
input_features = processor.feature_extractor(raw_speech=input_speech, return_tensors="pt").input_features.to(