Fix bug with SaveAudio node with --gpu-only

comfyanonymous · Jul 21, 2024 · f836e69 · f836e69
1 parent 5b69cfe
commit f836e69
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfy_extras/nodes_audio.py b/comfy_extras/nodes_audio.py
@@ -147,7 +147,7 @@ def save_audio(self, audio, filename_prefix="ComfyUI", prompt=None, extra_pnginf
                 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"