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

Change to Euler Ancestral

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- #from diffusers import EulerAncestralDiscreteScheduler
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
- #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
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