Skip to content

Commit

Permalink
fix Whisper tests on GPU (huggingface#23753)
Browse files Browse the repository at this point in the history
* move input features to GPU

* skip these tests because undefined behavior

* unskip tests
  • Loading branch information
hollance authored and sheonhan committed Jun 1, 2023
1 parent 76efb2a commit 79b003a
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/models/whisper/test_modeling_whisper.py
Original file line number Diff line number Diff line change
Expand Up @@ -1477,7 +1477,7 @@ def test_generate_with_prompt_ids(self):
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny")
model.to(torch_device)
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)
prompt_ids = processor.get_prompt_ids("Leighton")
Expand All @@ -1494,7 +1494,7 @@ def test_generate_with_prompt_ids_and_forced_decoder_ids(self):
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny")
model.to(torch_device)
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"
language = "de"
expected_tokens = [f"<|{task}|>", f"<|{language}|>"]
Expand All @@ -1513,7 +1513,7 @@ def test_generate_with_prompt_ids_and_no_non_prompt_forced_decoder_ids(self):
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny.en")
model.to(torch_device)
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_ids = processor.get_prompt_ids(prompt)

Expand Down

0 comments on commit 79b003a

Please sign in to comment.