wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-22 15:59:45 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Node replacement for audio nodes

Browse Source
This commit is contained in:
Alexis Rolland 2026-06-18 15:03:44 +08:00
parent f2270f070a
commit 99c33436d5
2 changed files with 40 additions and 99 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

97
comfy_extras/nodes_audio.py
View File

@ -152,105 +152,13 @@ class VAEDecodeAudioTiled(IO.ComfyNode):
return IO.NodeOutput(vae_decode_audio(vae, samples, tile_size, overlap))
class SaveAudio(IO.ComfyNode):
@classmethod
def define_schema(cls):
return IO.Schema(
node_id="SaveAudio",
search_aliases=["export flac"],
display_name="Save Audio (FLAC) (Deprecated)",
category="audio",
essentials_category="Audio",
inputs=[
IO.Audio.Input("audio"),
IO.String.Input("filename_prefix", default="audio/ComfyUI"),
],
hidden=[IO.Hidden.prompt, IO.Hidden.extra_pnginfo],
is_output_node=True,
is_deprecated=True,
)
@classmethod
def execute(cls, audio, filename_prefix="ComfyUI", format="flac") -> IO.NodeOutput:
if audio is None:
raise ValueError("SaveAudio: input audio is None (source video may have no audio track).")
return IO.NodeOutput(
ui=UI.AudioSaveHelper.get_save_audio_ui(audio, filename_prefix=filename_prefix, cls=cls, format=format)
)
save_flac = execute # TODO: remove
class SaveAudioMP3(IO.ComfyNode):
@classmethod
def define_schema(cls):
return IO.Schema(
node_id="SaveAudioMP3",
search_aliases=["export mp3"],
display_name="Save Audio (MP3) (Deprecated)",
category="audio",
essentials_category="Audio",
inputs=[
IO.Audio.Input("audio"),
IO.String.Input("filename_prefix", default="audio/ComfyUI"),
IO.Combo.Input("quality", options=["V0", "128k", "320k"], default="V0"),
],
hidden=[IO.Hidden.prompt, IO.Hidden.extra_pnginfo],
is_output_node=True,
is_deprecated=True,
)
@classmethod
def execute(cls, audio, filename_prefix="ComfyUI", format="mp3", quality="128k") -> IO.NodeOutput:
if audio is None:
raise ValueError("SaveAudioMP3: input audio is None (source video may have no audio track).")
return IO.NodeOutput(
ui=UI.AudioSaveHelper.get_save_audio_ui(
audio, filename_prefix=filename_prefix, cls=cls, format=format, quality=quality
)
)
save_mp3 = execute # TODO: remove
class SaveAudioOpus(IO.ComfyNode):
@classmethod
def define_schema(cls):
return IO.Schema(
node_id="SaveAudioOpus",
search_aliases=["export opus"],
display_name="Save Audio (Opus) (Deprecated)",
category="audio",
inputs=[
IO.Audio.Input("audio"),
IO.String.Input("filename_prefix", default="audio/ComfyUI"),
IO.Combo.Input("quality", options=["64k", "96k", "128k", "192k", "320k"], default="128k"),
],
hidden=[IO.Hidden.prompt, IO.Hidden.extra_pnginfo],
is_output_node=True,
is_deprecated=True,
)
@classmethod
def execute(cls, audio, filename_prefix="ComfyUI", format="opus", quality="V3") -> IO.NodeOutput:
if audio is None:
raise ValueError("SaveAudioOpus: input audio is None (source video may have no audio track).")
return IO.NodeOutput(
ui=UI.AudioSaveHelper.get_save_audio_ui(
audio, filename_prefix=filename_prefix, cls=cls, format=format, quality=quality
)
)
save_opus = execute # TODO: remove
class SaveAudioAdvanced(IO.ComfyNode):
@classmethod
def define_schema(cls):
return IO.Schema(
node_id="SaveAudioAdvanced",
search_aliases=["save audio", "export audio", "output audio", "write audio", "flac", "mp3", "opus"],
display_name="Save Audio (Advanced)",
display_name="Save Audio",
description="Saves the input audio to your ComfyUI output directory.",
category="audio",
inputs=[
@ -870,9 +778,6 @@ class AudioExtension(ComfyExtension):
VAEEncodeAudio,
VAEDecodeAudio,
VAEDecodeAudioTiled,
SaveAudio,
SaveAudioMP3,
SaveAudioOpus,
SaveAudioAdvanced,
LoadAudio,
PreviewAudio,

42
comfy_extras/nodes_replacements.py
View File

@ -5,14 +5,17 @@ api = ComfyAPI()
async def register_replacements():
"""Register all built-in node replacements."""
await register_replacements_longeredge()
await register_replacements_batchimages()
await register_replacements_upscaleimage()
await register_replacements_conditioningavg()
await register_replacements_controlnet()
await register_replacements_load3d()
await register_replacements_longeredge()
await register_replacements_preview3d()
await register_replacements_saveaudio()
await register_replacements_saveaudiomp3()
await register_replacements_saveaudioopus()
await register_replacements_svdimg2vid()
await register_replacements_conditioningavg()
await register_replacements_upscaleimage()
async def register_replacements_longeredge():
# No dynamic inputs here
@ -92,6 +95,39 @@ async def register_replacements_conditioningavg():
old_node_id="ConditioningAverage ",
))
async def register_replacements_saveaudio():
# Replace deprecated node SaveAudio
await api.node_replacement.register(io.NodeReplace(
new_node_id="SaveAudioAdvanced",
old_node_id="SaveAudio",
input_mapping=[
{"new_id": "audio", "old_id": "audio"},
{"new_id": "format", "set_value": "flac"}
]
))
async def register_replacements_saveaudiomp3():
# Replace deprecated node SaveAudioMP3
await api.node_replacement.register(io.NodeReplace(
new_node_id="SaveAudioAdvanced",
old_node_id="SaveAudioMP3",
input_mapping=[
{"new_id": "audio", "old_id": "audio"},
{"new_id": "format", "set_value": "mp3"}
]
))
async def register_replacements_saveaudioopus():
# Replace deprecated node SaveAudioOpus
await api.node_replacement.register(io.NodeReplace(
new_node_id="SaveAudioAdvanced",
old_node_id="SaveAudioOpus",
input_mapping=[
{"new_id": "audio", "old_id": "audio"},
{"new_id": "format", "set_value": "opus"}
]
))
class NodeReplacementsExtension(ComfyExtension):
async def on_load(self) -> None:
await register_replacements()