Fix bug with SaveAudio node with --gpu-only

7 months ago · f836e69346
parent 5b69cfe7c3
commit f836e69346
1 changed files with 1 additions and 1 deletions
--- a/comfy_extras/nodes_audio.py
+++ b/comfy_extras/nodes_audio.py
@ -147,7 +147,7 @@ class SaveAudio:
                for x in extra_pnginfo:
                    metadata[x] = json.dumps(extra_pnginfo[x])

-        for (batch_number, waveform) in enumerate(audio["waveform"]):
+        for (batch_number, waveform) in enumerate(audio["waveform"].cpu()):
            filename_with_batch_num = filename.replace("%batch_num%", str(batch_number))
            file = f"{filename_with_batch_num}_{counter:05}_.flac"