diff --git a/comfy_extras/nodes_audio.py b/comfy_extras/nodes_audio.py index 6990b3f9..1c763f25 100644 --- a/comfy_extras/nodes_audio.py +++ b/comfy_extras/nodes_audio.py @@ -58,6 +58,9 @@ def INPUT_TYPES(s): def decode(self, vae, samples): audio = vae.decode(samples["samples"]).movedim(-1, 1) + std = torch.std(audio, dim=[1,2], keepdim=True) * 5.0 + std[std < 1.0] = 1.0 + audio /= std return ({"waveform": audio, "sample_rate": 44100}, )