xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-11 17:06:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

add wav file format option to save_audio

Browse Source
This commit is contained in:
Yuma Lenz 2024-08-31 21:58:17 +02:00
parent 1184ff6008
commit c12e3fe5da
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy_extras/nodes_audio.py
View File

@ -123,7 +123,8 @@ class SaveAudio:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "audio": ("AUDIO", ), return {"required": { "audio": ("AUDIO", ),
"filename_prefix": ("STRING", {"default": "audio/ComfyUI"})}, "filename_prefix": ("STRING", {"default": "audio/ComfyUI"}),
"file_format": (["wav", "flac"],{"default": "flac"})},
"hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"}, "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
} }
@ -134,7 +135,7 @@ class SaveAudio:
CATEGORY = "audio" CATEGORY = "audio"
def save_audio(self, audio, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=None): def save_audio(self, audio, filename_prefix="ComfyUI", file_format="flac", prompt=None, extra_pnginfo=None):
filename_prefix += self.prefix_append filename_prefix += self.prefix_append
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir) full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir)
results = list() results = list()
@ -149,12 +150,13 @@ class SaveAudio:
for (batch_number, waveform) in enumerate(audio["waveform"].cpu()): for (batch_number, waveform) in enumerate(audio["waveform"].cpu()):
filename_with_batch_num = filename.replace("%batch_num%", str(batch_number)) filename_with_batch_num = filename.replace("%batch_num%", str(batch_number))
file = f"{filename_with_batch_num}_{counter:05}_.flac" file = f"{filename_with_batch_num}_{counter:05}_.{file_format}"
buff = io.BytesIO() buff = io.BytesIO()
torchaudio.save(buff, waveform, audio["sample_rate"], format="FLAC") torchaudio.save(buff, waveform, audio["sample_rate"], format=file_format)
buff = insert_or_replace_vorbis_comment(buff, metadata) if (file_format == "flac"):
buff = insert_or_replace_vorbis_comment(buff, metadata)
with open(os.path.join(full_output_folder, file), 'wb') as f: with open(os.path.join(full_output_folder, file), 'wb') as f:
f.write(buff.getbuffer()) f.write(buff.getbuffer())