From c524fd8589338e9d350af61381cfe2686ccf76e3 Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Mon, 5 Dec 2022 19:37:46 +0000
Subject: [PATCH] correct librosa import

---
 src/diffusers/pipelines/audio_diffusion/mel.py | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/diffusers/pipelines/audio_diffusion/mel.py b/src/diffusers/pipelines/audio_diffusion/mel.py
index f9ac5654b4..fbc756d496 100644
--- a/src/diffusers/pipelines/audio_diffusion/mel.py
+++ b/src/diffusers/pipelines/audio_diffusion/mel.py
@@ -23,7 +23,19 @@ warnings.filterwarnings("ignore")
 
 import numpy as np  # noqa: E402
 
-import librosa  # noqa: E402
+
+try:
+    import librosa  # noqa: E402
+
+    _librosa_can_be_imported = True
+    _import_error = ""
+except Exception as e:
+    _librosa_can_be_imported = False
+    _import_error = (
+        f"Cannot import librosa because {e}. Make sure to correctly install librosa to be able to install it."
+    )
+
+
 from PIL import Image  # noqa: E402
 
 
@@ -60,6 +72,9 @@ class Mel(ConfigMixin, SchedulerMixin):
         self.set_resolution(x_res, y_res)
         self.audio = None
 
+        if not _librosa_can_be_imported:
+            raise ValueError(_import_error)
+
     def set_resolution(self, x_res: int, y_res: int):
         """Set resolution.