Added safety around torchaudio import in _ui.py

comfy_api/latest/_ui.py

@@ -9,7 +9,12 @@ from typing import Type
 import av
 import numpy as np
 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.PngImagePlugin import PngInfo
 
@@ -302,6 +307,8 @@ class AudioSaveHelper:
 
                 # Resample if necessary
                 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)
 
             # Create output with specified format