xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-05 07:17:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix ruff

Browse Source
This commit is contained in:
thot-experiment 2025-05-09 00:38:41 -07:00
parent 6554a96d20
commit 269d141ce7
1 changed files with 17 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes_audio.py
View File

@ -1,7 +1,6 @@
from __future__ import annotations from __future__ import annotations
import av import av
import tempfile
import torchaudio import torchaudio
import torch import torch
import comfy.model_management import comfy.model_management
@ -167,9 +166,7 @@ class SaveAudio:
CATEGORY = "audio" CATEGORY = "audio"
def save_audio(self, audio, filename_prefix="ComfyUI", format="flac", prompt=None, extra_pnginfo=None): def save_audio(self, audio, filename_prefix="ComfyUI", format="flac", prompt=None, extra_pnginfo=None):
import av
import io import io
import numpy as np
filename_prefix += self.prefix_append filename_prefix += self.prefix_append
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir) full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir)
@ -229,7 +226,7 @@ class SaveAudio:
output_container.metadata[key] = value output_container.metadata[key] = value
# Set up the output stream with appropriate properties # Set up the output stream with appropriate properties
in_stream = input_container.streams.audio[0] input_container.streams.audio[0]
if format == "opus": if format == "opus":
out_stream = output_container.add_stream("libopus", rate=sample_rate) out_stream = output_container.add_stream("libopus", rate=sample_rate)
elif format == "mp3": elif format == "mp3":