Update app.py
Browse filesAdded torch generator so I can seed it
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.")
|