"...resnet50_tensorflow.git" did not exist on "34d91a7c1591f5b714d2c17515977f57d3652004"
Unverified Commit aade711d authored by Fanli Lin's avatar Fanli Lin Committed by GitHub
Browse files

[tests] enable automatic speech recognition pipeline tests on XPU (#29308)

* use require_torch_gpu

* enable on XPU
parent 831bc25d
...@@ -1221,7 +1221,7 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase): ...@@ -1221,7 +1221,7 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase):
processor = AutoProcessor.from_pretrained("openai/whisper-tiny.en") processor = AutoProcessor.from_pretrained("openai/whisper-tiny.en")
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny.en") model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny.en")
model = model.to("cuda") model = model.to(torch_device)
pipe = pipeline( pipe = pipeline(
"automatic-speech-recognition", "automatic-speech-recognition",
...@@ -1229,7 +1229,7 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase): ...@@ -1229,7 +1229,7 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase):
tokenizer=processor.tokenizer, tokenizer=processor.tokenizer,
feature_extractor=processor.feature_extractor, feature_extractor=processor.feature_extractor,
max_new_tokens=128, max_new_tokens=128,
device="cuda:0", device=torch_device,
) )
ds = load_dataset("distil-whisper/meanwhile", "default")["test"] ds = load_dataset("distil-whisper/meanwhile", "default")["test"]
...@@ -1246,7 +1246,7 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase): ...@@ -1246,7 +1246,7 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase):
pipe = pipeline( pipe = pipeline(
"automatic-speech-recognition", "automatic-speech-recognition",
model="facebook/seamless-m4t-v2-large", model="facebook/seamless-m4t-v2-large",
device="cuda:0", device=torch_device,
) )
dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation") dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment