1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

correct librosa import

This commit is contained in:
Patrick von Platen
2022-12-05 19:37:46 +00:00
parent 62b497c418
commit c524fd8589

View File

@@ -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.