jytole commited on
Commit
52a5324
1 Parent(s): ef79c00

Update app.py

Browse files

Added torch generator so I can seed it

Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,6 +9,8 @@ pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
9
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
10
  pipe = pipe.to("cpu")
11
 
 
 
12
  def texttoaudio(prompt, neg_prompt, seed, inf_steps, guidance_scale):
13
  if prompt is None:
14
  raise gr.Error("Please provide a text input.")
 
9
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
10
  pipe = pipe.to("cpu")
11
 
12
+ generator = torch.Generator("cpu")
13
+
14
  def texttoaudio(prompt, neg_prompt, seed, inf_steps, guidance_scale):
15
  if prompt is None:
16
  raise gr.Error("Please provide a text input.")