tmp commit

huggingface · Dec 15, 2023 · 11dace1 · 11dace1
1 parent 7d99f84
commit 11dace1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/whisper/test_modeling_whisper.py b/tests/models/whisper/test_modeling_whisper.py
@@ -1651,7 +1651,7 @@ def test_large_generation_multilingual(self):
         model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large")
         model.to(torch_device)
 
-        ds = load_dataset("mozilla-foundation/common_voice_11_0", "ja", split="test", streaming=True)
+        ds = load_dataset("common_voice", "ja", split="test", streaming=True)
         ds = ds.cast_column("audio", datasets.Audio(sampling_rate=16_000))
         input_speech = next(iter(ds))["audio"]["array"]
         input_features = processor.feature_extractor(raw_speech=input_speech, return_tensors="pt").input_features.to(