mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-14 00:56:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

_ui.py import torchaudio safety check (#9234)

Browse Source 
* Added safety around torchaudio import in _ui.py

* Trusted cursor too much, fixed torchaudio bool
This commit is contained in:
Jedrzej Kosinski 2025-08-07 14:54:09 -07:00 committed by GitHub
parent 05df2df489
commit 42974a448c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_api/latest/_ui.py
View File

@ -9,7 +9,11 @@ from typing import Type
import av import av
import numpy as np import numpy as np
import torch import torch
import torchaudio try:
import torchaudio
TORCH_AUDIO_AVAILABLE = True
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 +306,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