From 3b35aa78467e515868ed56f587db1506d98d50dd Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Tue, 18 Jun 2024 14:44:57 -0700
Subject: [PATCH] For now, make torchaudio optional, because this package
 requires a lot of work to prevent weird dependency issues

---
 comfy_extras/nodes/nodes_audio.py | 5 ++++-
 requirements.txt                  | 1 -
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/comfy_extras/nodes/nodes_audio.py b/comfy_extras/nodes/nodes_audio.py
index a3f3650ed..566fd5c99 100644
--- a/comfy_extras/nodes/nodes_audio.py
+++ b/comfy_extras/nodes/nodes_audio.py
@@ -1,4 +1,3 @@
-import torchaudio
 import torch
 import comfy.model_management
 from comfy.cmd import folder_paths
@@ -69,6 +68,8 @@ class SaveAudio:
     CATEGORY = "_for_testing/audio"
 
     def save_audio(self, audio, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=None):
+        import torchaudio
+
         filename_prefix += self.prefix_append
         full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir)
         results = list()
@@ -99,6 +100,8 @@ class LoadAudio:
     FUNCTION = "load"
 
     def load(self, audio):
+        import torchaudio
+
         audio_path = folder_paths.get_annotated_filepath(audio)
         waveform, sample_rate = torchaudio.load(audio_path)
         multiplier = 1.0
diff --git a/requirements.txt b/requirements.txt
index 521ccd144..fd6e65921 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,5 @@
 torch
 torchvision
-torchaudio
 torchdiffeq>=0.2.3
 torchsde>=0.2.6
 einops>=0.6.0