[Speech] Move all examples to new audio feature (#14045)
* up * up * up * finish
This commit is contained in:
committed by
GitHub
parent
4334095c32
commit
bdf31d6e0a
@@ -407,21 +407,13 @@ class SEWModelIntegrationTest(unittest.TestCase):
|
||||
def _load_datasamples(self, num_samples):
|
||||
from datasets import load_dataset
|
||||
|
||||
import soundfile as sf
|
||||
ds = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
|
||||
# automatic decoding with librispeech
|
||||
speech_samples = ds.sort("id").filter(
|
||||
lambda x: x["id"] in [f"1272-141231-000{i}" for i in range(num_samples)]
|
||||
)[:num_samples]["audio"]
|
||||
|
||||
ids = [f"1272-141231-000{i}" for i in range(num_samples)]
|
||||
|
||||
# map files to raw
|
||||
def map_to_array(batch):
|
||||
speech, _ = sf.read(batch["file"])
|
||||
batch["speech"] = speech
|
||||
return batch
|
||||
|
||||
ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
|
||||
|
||||
ds = ds.filter(lambda x: x["id"] in ids).sort("id").map(map_to_array)
|
||||
|
||||
return ds["speech"][:num_samples]
|
||||
return [x["array"] for x in speech_samples]
|
||||
|
||||
def test_inference_pretrained_batched(self):
|
||||
model = SEWModel.from_pretrained("asapp/sew-tiny-100k").to(torch_device)
|
||||
|
||||
Reference in New Issue
Block a user