Function save_batch_audio

tests/stable_audio/stable_audio_python_warm_bench.py:300–307

(audio: torch.Tensor, output_dir: Path, sample_rate: int)

def save_batch_audio(audio: torch.Tensor, output_dir: Path, sample_rate: int) -> list[dict[str, Any]]:
    output_dir.mkdir(parents=True, exist_ok=True)
    stems: list[dict[str, Any]] = []
    for index in range(audio.shape[0]):
        path = output_dir / ("audio.wav" if audio.shape[0] == 1 else f"audio_{index:02d}.wav")
        torchaudio.save(str(path), audio[index].detach().cpu(), sample_rate, encoding="PCM_F")
        stems.append({"name": f"audio_{index}", "audio": str(path), "summary": summarize_audio(path)})
    return stems


def run_request(