teticio commited on
Commit
52ca0bd
1 Parent(s): 6456a7f

temporary patch

Browse files
Files changed (1) hide show
  1. scripts/train_unconditional.py +1 -2
scripts/train_unconditional.py CHANGED
@@ -11,8 +11,6 @@ from accelerate import Accelerator
11
  from accelerate.logging import get_logger
12
  from datasets import load_from_disk, load_dataset
13
  from diffusers import (
14
- AudioDiffusionPipeline,
15
- Mel,
16
  DDPMScheduler,
17
  UNet2DModel,
18
  DDIMScheduler,
@@ -29,6 +27,7 @@ from torchvision.transforms import (
29
  import numpy as np
30
  from tqdm.auto import tqdm
31
  from librosa.util import normalize
 
32
 
33
  logger = get_logger(__name__)
34
 
 
11
  from accelerate.logging import get_logger
12
  from datasets import load_from_disk, load_dataset
13
  from diffusers import (
 
 
14
  DDPMScheduler,
15
  UNet2DModel,
16
  DDIMScheduler,
 
27
  import numpy as np
28
  from tqdm.auto import tqdm
29
  from librosa.util import normalize
30
+ from audiodiffusion import AudioDiffusionPipeline, Mel
31
 
32
  logger = get_logger(__name__)
33