fix(inference): inference crashes on non-audio files in (+1 more)

[andrewwhitecdw]

This PR addresses 2 issues in inference.py.

Fix 1

fix: inference crashes on non-audio files in input directory

Fix: Replace:

    filelist = os.listdir(a.input_wavs_dir)

with:

    filelist = [
        f
        for f in os.listdir(a.input_wavs_dir)
        if f.lower().endswith((".wav", ".flac", ".mp3"))
    ]

Fix 2

fix: int16 cast can wrap around at full scale

Fix: Replace:

            audio = audio * MAX_WAV_VALUE
            audio = audio.cpu().numpy().astype("int16")

with:

            audio = audio * MAX_WAV_VALUE
            audio = audio.clamp(min=-MAX_WAV_VALUE, max=MAX_WAV_VALUE - 1)
            audio = audio.cpu().numpy().astype("int16")

Files changed

• inference.py