Fix bug with SaveAudio node with --gpu-only

This commit is contained in:
comfyanonymous 2024-07-21 16:16:45 -04:00
parent 5b69cfe7c3
commit f836e69346

View File

@ -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"