mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-31 08:40:19 +08:00
feat(LoadAudio): add duration output
This commit is contained in:
parent
7ee77ff038
commit
98756ecf5e
@ -271,7 +271,10 @@ class LoadAudio(IO.ComfyNode):
|
|||||||
inputs=[
|
inputs=[
|
||||||
IO.Combo.Input("audio", upload=IO.UploadType.audio, options=sorted(files)),
|
IO.Combo.Input("audio", upload=IO.UploadType.audio, options=sorted(files)),
|
||||||
],
|
],
|
||||||
outputs=[IO.Audio.Output()],
|
outputs=[
|
||||||
|
IO.Audio.Output(),
|
||||||
|
IO.Float.Output(display_name="duration"),
|
||||||
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@ -279,7 +282,7 @@ class LoadAudio(IO.ComfyNode):
|
|||||||
audio_path = folder_paths.get_annotated_filepath(audio)
|
audio_path = folder_paths.get_annotated_filepath(audio)
|
||||||
waveform, sample_rate = load(audio_path)
|
waveform, sample_rate = load(audio_path)
|
||||||
audio = {"waveform": waveform.unsqueeze(0), "sample_rate": sample_rate}
|
audio = {"waveform": waveform.unsqueeze(0), "sample_rate": sample_rate}
|
||||||
return IO.NodeOutput(audio)
|
return IO.NodeOutput(audio, waveform.shape[-1] / sample_rate)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def fingerprint_inputs(cls, audio):
|
def fingerprint_inputs(cls, audio):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user