Change to VQDiffusion

app.py

@@ -18,8 +18,8 @@ from diffusers import AudioLDMPipeline
 #from diffusers import ScoreSdeVeScheduler
 #from diffusers import IPNDMScheduler
 #from diffusers import EulerDiscreteScheduler
-from diffusers import EulerAncestralDiscreteScheduler
-#from diffusers import VQDiffusionScheduler
+#from diffusers import EulerAncestralDiscreteScheduler
+from diffusers import VQDiffusionScheduler
 #from diffusers import UniPCMultistepScheduler
 #from diffusers import RePaintScheduler
 
@@ -47,8 +47,8 @@ pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
 #pipe.scheduler = ScoreSdeVeScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = IPNDMScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
-pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-#pipe.scheduler = VQDiffusionScheduler.from_config(pipe.scheduler.config)
+#pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+pipe.scheduler = VQDiffusionScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = RePaintScheduler.from_config(pipe.scheduler.config)
 pipe = pipe.to(device)