fix for latest version of diffusers

audiodiffusion/__init__.py

@@ -11,7 +11,7 @@ from diffusers import (DiffusionPipeline, UNet2DConditionModel, DDIMScheduler,
 
 from .mel import Mel
 
-VERSION = "1.2.3"
+VERSION = "1.2.4"
 
 
 class AudioDiffusion:

scripts/train_unconditional.py

@@ -126,10 +126,10 @@ def main(args):
 
     if args.scheduler == "ddpm":
         noise_scheduler = DDPMScheduler(
-            num_train_timesteps=args.num_train_steps, tensor_format="pt")
+            num_train_timesteps=args.num_train_steps)
     else:
         noise_scheduler = DDIMScheduler(
-            num_train_timesteps=args.num_train_steps, tensor_format="pt")
+            num_train_timesteps=args.num_train_steps)
 
     optimizer = torch.optim.AdamW(
         model.parameters(),