diff --git a/comfy_extras/nodes/nodes_audio.py b/comfy_extras/nodes/nodes_audio.py index 7a4429b87..8ab3541ed 100644 --- a/comfy_extras/nodes/nodes_audio.py +++ b/comfy_extras/nodes/nodes_audio.py @@ -40,7 +40,7 @@ class VAEEncodeAudio: def encode(self, vae, audio): sample_rate = audio["sample_rate"] if 44100 != sample_rate: - import torchaudio + import torchaudio # pylint: disable=import-error waveform = torchaudio.functional.resample(audio["waveform"], sample_rate, 44100) else: waveform = audio["waveform"] @@ -137,7 +137,7 @@ class SaveAudio: CATEGORY = "_for_testing/audio" 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 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" def load(self, audio): - import torchaudio + import torchaudio # pylint: disable=import-error audio_path = folder_paths.get_annotated_filepath(audio) waveform, sample_rate = torchaudio.load(audio_path) diff --git a/tests/conftest.py b/tests/conftest.py index 6727711a9..9972e77aa 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -54,6 +54,7 @@ def has_gpu() -> bool: try: import intel_extension_for_pytorch as ipex has_gpu = ipex.xpu.device_count() > 0 + return has_gpu except ImportError: try: import torch