fix for latest version of diffusers

audiodiffusion/__init__.py

@@ -9,7 +9,7 @@ from tqdm.auto import tqdm
 # from diffusers import AudioDiffusionPipeline
 from .pipeline_audio_diffusion import AudioDiffusionPipeline
 
-VERSION = "1.4.2"
+VERSION = "1.5.0"
 
 
 class AudioDiffusion:

audiodiffusion/pipeline_audio_diffusion.py

@@ -21,8 +21,16 @@ from typing import List, Tuple, Union
 
 import numpy as np
 import torch
-from diffusers import AutoencoderKL, DDIMScheduler, DDPMScheduler, UNet2DConditionModel
-from diffusers.pipeline_utils import AudioPipelineOutput, BaseOutput, DiffusionPipeline, ImagePipelineOutput
+from diffusers import (
+    AudioPipelineOutput,
+    AutoencoderKL,
+    DDIMScheduler,
+    DDPMScheduler,
+    DiffusionPipeline,
+    ImagePipelineOutput,
+    UNet2DConditionModel,
+)
+from diffusers.utils import BaseOutput
 from PIL import Image
 
 from .mel import Mel

requirements.txt

@@ -1,7 +1,7 @@
 torch
 numpy
 Pillow
-diffusers>=0.10.0
+diffusers>=0.12.0
 librosa
 datasets
 gradio

setup.cfg

@@ -15,6 +15,6 @@ install_requires =
     torch
     numpy
     Pillow
-    diffusers>=0.10.0
+    diffusers>=0.12.0
     librosa
     datasets