Function load_audio

tests/stable_audio/stable_audio_python_warm_bench.py:295–297

(path: Path)

def load_audio(path: Path) -> tuple[int, torch.Tensor]:
    waveform, sample_rate = torchaudio.load(str(path))
    return sample_rate, waveform


def save_batch_audio(audio: torch.Tensor, output_dir: Path, sample_rate: int) -> list[dict[str, Any]]: