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.")
|