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

fix Whisper tests on GPU (#23753)

Browse Source 
* move input features to GPU

* skip these tests because undefined behavior

* unskip tests
This commit is contained in:
Matthijs Hollemans
2023-05-30 15:06:58 +02:00
committed by GitHub
parent ac224dee90
commit 2faa09530b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1477,7 +1477,7 @@ class WhisperModelIntegrationTests(unittest.TestCase):
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny") model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny")
model.to(torch_device) model.to(torch_device)
input_speech = self._load_datasamples(4)[-1:] input_speech = self._load_datasamples(4)[-1:]
input_features = processor(input_speech, return_tensors="pt").input_features input_features = processor(input_speech, return_tensors="pt").input_features.to(torch_device)
output_without_prompt = model.generate(input_features) output_without_prompt = model.generate(input_features)
prompt_ids = processor.get_prompt_ids("Leighton") prompt_ids = processor.get_prompt_ids("Leighton")
@@ -1494,7 +1494,7 @@ class WhisperModelIntegrationTests(unittest.TestCase):
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny") model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny")
model.to(torch_device) model.to(torch_device)
input_speech = self._load_datasamples(1) input_speech = self._load_datasamples(1)
input_features = processor(input_speech, return_tensors="pt").input_features input_features = processor(input_speech, return_tensors="pt").input_features.to(torch_device)
task = "translate" task = "translate"
language = "de" language = "de"
expected_tokens = [f"<|{task}|>", f"<|{language}|>"] expected_tokens = [f"<|{task}|>", f"<|{language}|>"]
@@ -1513,7 +1513,7 @@ class WhisperModelIntegrationTests(unittest.TestCase):
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny.en") model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny.en")
model.to(torch_device) model.to(torch_device)
input_speech = self._load_datasamples(1) input_speech = self._load_datasamples(1)
input_features = processor(input_speech, return_tensors="pt").input_features input_features = processor(input_speech, return_tensors="pt").input_features.to(torch_device)
prompt = "test prompt" prompt = "test prompt"
prompt_ids = processor.get_prompt_ids(prompt) prompt_ids = processor.get_prompt_ids(prompt)