xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-28 00:12:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added safety around torchaudio import in _ui.py

Browse Source
This commit is contained in:
Jedrzej Kosinski 2025-08-07 14:39:42 -07:00
parent 05df2df489
commit 982965f872
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy_api/latest/_ui.py
View File

@ -9,7 +9,12 @@ from typing import Type
import av import av
import numpy as np import numpy as np
import torch import torch
import torchaudio TORCH_AUDIO_AVAILABLE = False
if TORCH_AUDIO_AVAILABLE:
try:
import torchaudio
except ImportError:
TORCH_AUDIO_AVAILABLE = False
from PIL import Image as PILImage from PIL import Image as PILImage
from PIL.PngImagePlugin import PngInfo from PIL.PngImagePlugin import PngInfo
@ -302,6 +307,8 @@ class AudioSaveHelper:
# Resample if necessary # Resample if necessary
if sample_rate != audio["sample_rate"]: if sample_rate != audio["sample_rate"]:
if not TORCH_AUDIO_AVAILABLE:
raise Exception("torchaudio is not available; cannot resample audio.")
waveform = torchaudio.functional.resample(waveform, audio["sample_rate"], sample_rate) waveform = torchaudio.functional.resample(waveform, audio["sample_rate"], sample_rate)
# Create output with specified format # Create output with specified format