wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-28 15:20:25 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix pylint issues; fix IPEX workflow run

Browse Source
This commit is contained in:
doctorpangloss 2024-07-05 12:44:03 -07:00
parent 5fef1e48f6
commit 7c6d716541
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_extras/nodes/nodes_audio.py
View File

@ -40,7 +40,7 @@ class VAEEncodeAudio:
def encode(self, vae, audio): def encode(self, vae, audio):
sample_rate = audio["sample_rate"] sample_rate = audio["sample_rate"]
if 44100 != sample_rate: if 44100 != sample_rate:
import torchaudio import torchaudio # pylint: disable=import-error
waveform = torchaudio.functional.resample(audio["waveform"], sample_rate, 44100) waveform = torchaudio.functional.resample(audio["waveform"], sample_rate, 44100)
else: else:
waveform = audio["waveform"] waveform = audio["waveform"]
@ -137,7 +137,7 @@ class SaveAudio:
CATEGORY = "_for_testing/audio" CATEGORY = "_for_testing/audio"
def save_audio(self, audio, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=None): def save_audio(self, audio, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=None):
import torchaudio import torchaudio # pylint: disable=import-error
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)
@ -205,7 +205,7 @@ class LoadAudio:
FUNCTION = "load" FUNCTION = "load"
def load(self, audio): def load(self, audio):
import torchaudio import torchaudio # pylint: disable=import-error
audio_path = folder_paths.get_annotated_filepath(audio) audio_path = folder_paths.get_annotated_filepath(audio)
waveform, sample_rate = torchaudio.load(audio_path) waveform, sample_rate = torchaudio.load(audio_path)

1
tests/conftest.py
View File

@ -54,6 +54,7 @@ def has_gpu() -> bool:
try: try:
import intel_extension_for_pytorch as ipex import intel_extension_for_pytorch as ipex
has_gpu = ipex.xpu.device_count() > 0 has_gpu = ipex.xpu.device_count() > 0
return has_gpu
except ImportError: except ImportError:
try: try:
import torch import torch