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

Add bit_depth option to SaveAudio for 16-bit/24-bit FLAC support

Browse Source
This commit is contained in:
MuziekMagie 2025-04-17 15:20:06 +02:00
parent 0d720e4367
commit 0191baf8f3
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy_extras/nodes_audio.py
View File

@ -153,7 +153,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"}),
"bit_depth": ([16, 24], )},
"hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"}, "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
} }
@ -164,7 +165,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", bit_depth=16, 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[FileLocator] = [] results: list[FileLocator] = []
@ -182,7 +183,7 @@ class SaveAudio:
file = f"{filename_with_batch_num}_{counter:05}_.flac" file = f"{filename_with_batch_num}_{counter:05}_.flac"
buff = io.BytesIO() buff = io.BytesIO()
torchaudio.save(buff, waveform, audio["sample_rate"], format="FLAC") torchaudio.save(buff, waveform, audio["sample_rate"], format="FLAC", bits_per_sample=bit_depth)
buff = insert_or_replace_vorbis_comment(buff, metadata) buff = insert_or_replace_vorbis_comment(buff, metadata)