temporary patch

teticio · Dec 5, 2022 · 52ca0bd · 52ca0bd
1 parent 6456a7f
commit 52ca0bd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/scripts/train_unconditional.py b/scripts/train_unconditional.py
@@ -11,8 +11,6 @@
 from accelerate.logging import get_logger
 from datasets import load_from_disk, load_dataset
 from diffusers import (
-    AudioDiffusionPipeline,
-    Mel,
     DDPMScheduler,
     UNet2DModel,
     DDIMScheduler,
@@ -29,6 +27,7 @@
 import numpy as np
 from tqdm.auto import tqdm
 from librosa.util import normalize
+from audiodiffusion import AudioDiffusionPipeline, Mel
 
 logger = get_logger(__name__)