fix: save_audio can write distorted audio due to missing clamp

[andrewwhitecdw]

Problem

fix: save_audio can write distorted audio due to missing clamp

Fix

Replace:

def save_audio(audio, path, sr):
    # wav: torch with 1d shape
    audio = audio * MAX_WAV_VALUE
    audio = audio.cpu().numpy().astype("int16")
    write(path, sr, audio)

with:

def save_audio(audio, path, sr):
    # wav: torch with 1d shape
    audio = audio.clamp(min=-1, max=1)
    audio = audio * MAX_WAV_VALUE
    audio = audio.cpu().numpy().astype("int16")
    write(path, sr, audio)

Files changed

• utils.py