[Seamless v2] Add FE to auto mapping (#27829)
This commit is contained in:
@@ -1115,6 +1115,23 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase):
|
||||
|
||||
assert result == EXPECTED_RESULT
|
||||
|
||||
@require_torch
|
||||
@slow
|
||||
def test_seamless_v2(self):
|
||||
pipe = pipeline(
|
||||
"automatic-speech-recognition",
|
||||
model="facebook/seamless-m4t-v2-large",
|
||||
device="cuda:0",
|
||||
)
|
||||
|
||||
dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
|
||||
sample = dataset[0]["audio"]
|
||||
|
||||
result = pipe(sample, generate_kwargs={"tgt_lang": "eng"})
|
||||
EXPECTED_RESULT = "mister quilter is the apostle of the middle classes and we are glad to welcome his gospel"
|
||||
|
||||
assert result["text"] == EXPECTED_RESULT
|
||||
|
||||
@require_torch
|
||||
@slow
|
||||
def test_chunking_and_timestamps(self):
|
||||
|
||||
Reference in New Issue
Block a user