jytole commited on
Commit
85c877d
1 Parent(s): 727bfb7

Change to UniPCMultistep Scheduler

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- #from diffusers import UniPCMultistepScheduler
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
- #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)
 
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)