Change to Euler Ancestral
Browse files
app.py
CHANGED
@@ -13,8 +13,8 @@ from diffusers import AudioLDMPipeline
|
|
13 |
#from diffusers import KDPM2AncestralDiscreteScheduler
|
14 |
#from diffusers import LMSDiscreteScheduler
|
15 |
#from diffusers import PNDMScheduler
|
16 |
-
from diffusers import EulerDiscreteScheduler
|
17 |
-
|
18 |
#from diffusers import UniPCMultistepScheduler
|
19 |
|
20 |
from transformers import AutoProcessor, ClapModel
|
@@ -37,8 +37,8 @@ pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
|
|
37 |
#pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
38 |
#pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
|
39 |
#pipe.scheduler = PNDMScheduler.from_config(pipe.scheduler.config)
|
40 |
-
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
|
41 |
-
|
42 |
#pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
43 |
pipe = pipe.to(device)
|
44 |
|
|
|
13 |
#from diffusers import KDPM2AncestralDiscreteScheduler
|
14 |
#from diffusers import LMSDiscreteScheduler
|
15 |
#from diffusers import PNDMScheduler
|
16 |
+
#from diffusers import EulerDiscreteScheduler
|
17 |
+
from diffusers import EulerAncestralDiscreteScheduler
|
18 |
#from diffusers import UniPCMultistepScheduler
|
19 |
|
20 |
from transformers import AutoProcessor, ClapModel
|
|
|
37 |
#pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
38 |
#pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
|
39 |
#pipe.scheduler = PNDMScheduler.from_config(pipe.scheduler.config)
|
40 |
+
#pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
|
41 |
+
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
42 |
#pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
43 |
pipe = pipe.to(device)
|
44 |
|