Change to UniPCMultistep Scheduler
Browse files
app.py
CHANGED
@@ -14,8 +14,8 @@ from diffusers import AudioLDMPipeline
|
|
14 |
#from diffusers import LMSDiscreteScheduler
|
15 |
#from diffusers import PNDMScheduler
|
16 |
#from diffusers import EulerDiscreteScheduler
|
17 |
-
from diffusers import EulerAncestralDiscreteScheduler
|
18 |
-
|
19 |
|
20 |
from transformers import AutoProcessor, ClapModel
|
21 |
|
@@ -38,8 +38,8 @@ pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
|
|
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 |
-
|
43 |
pipe = pipe.to(device)
|
44 |
|
45 |
clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)
|
|
|
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
|
21 |
|
|
|
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 |
|
45 |
clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)
|