qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Resample audio to 44100 when VAE encoding it.

This commit is contained in:
comfyanonymous 2024-06-24 16:55:20 -04:00
parent 866f54da8d
commit eab211bb1e
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_audio.py
View File

@ -31,7 +31,13 @@ class VAEEncodeAudio:
CATEGORY = "_for_testing/audio" CATEGORY = "_for_testing/audio"
def encode(self, vae, audio): def encode(self, vae, audio):
t = vae.encode(audio["waveform"].movedim(1, -1)) sample_rate = audio["sample_rate"]
if 44100 != sample_rate:
waveform = torchaudio.functional.resample(audio["waveform"], sample_rate, 44100)
else:
waveform = audio["waveform"]
t = vae.encode(waveform.movedim(1, -1))
return ({"samples":t}, ) return ({"samples":t}, )
class VAEDecodeAudio: class VAEDecodeAudio: