From f48bf5d62965620e5e76efc8c82d33928cf18f28 Mon Sep 17 00:00:00 2001 From: Miro <200482516+Mirochill@users.noreply.github.com> Date: Wed, 27 May 2026 11:16:56 +0200 Subject: [PATCH] Use ffmpeg backend for MP3 audio saves --- nodes/audio_nodes.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/nodes/audio_nodes.py b/nodes/audio_nodes.py index 5642cae..038b6f4 100644 --- a/nodes/audio_nodes.py +++ b/nodes/audio_nodes.py @@ -57,14 +57,11 @@ class VrchAudioSaverNode: elif waveform.dim() > 2: waveform = waveform.view(waveform.size(0), -1) - # Save audio using BytesIO - buff = io.BytesIO() - torchaudio.save(buff, waveform, audio["sample_rate"], format=extension.upper()) - buff.seek(0) - - # Write BytesIO content to file - with open(full_path, 'wb') as f: - f.write(buff.getvalue()) + audio_format = extension.lower() + save_kwargs = {"format": audio_format} + if audio_format == "mp3": + save_kwargs["backend"] = "ffmpeg" + torchaudio.save(full_path, waveform, audio["sample_rate"], **save_kwargs) results.append({ "filename": file_name,