[ci-daily] Fix pipeline tests (#21257)
* use streaming dataset * fix whisper's test * add rescale argument to chunk_iter
This commit is contained in:
@@ -201,8 +201,9 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase, metaclass=Pipel
|
||||
@require_torch
|
||||
@require_pyctcdecode
|
||||
def test_large_model_pt_with_lm(self):
|
||||
dataset = load_dataset("Narsil/asr_dummy")
|
||||
filename = dataset["test"][3]["file"]
|
||||
dataset = load_dataset("Narsil/asr_dummy", streaming=True)
|
||||
third_item = next(iter(dataset["test"].skip(3)))
|
||||
filename = third_item["file"]
|
||||
|
||||
speech_recognizer = pipeline(
|
||||
task="automatic-speech-recognition",
|
||||
|
||||
Reference in New Issue
Block a user