Spaces:
Runtime error
Runtime error
vaibhavs10
commited on
Commit
Β·
f04e297
1
Parent(s):
bf58ccb
up
Browse files
app.py
CHANGED
|
@@ -5,7 +5,9 @@ import torch
|
|
| 5 |
|
| 6 |
from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
|
| 7 |
|
| 8 |
-
pipe = SpectrogramDiffusionPipeline.from_pretrained(
|
|
|
|
|
|
|
| 9 |
pipe.enable_xformers_memory_efficient_attention()
|
| 10 |
|
| 11 |
processor = MidiProcessor()
|
|
@@ -41,6 +43,6 @@ gr.Interface(
|
|
| 41 |
],
|
| 42 |
title=title,
|
| 43 |
description=description,
|
| 44 |
-
theme=
|
| 45 |
examples=examples,
|
| 46 |
-
).launch(debug=True)
|
|
|
|
| 5 |
|
| 6 |
from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
|
| 7 |
|
| 8 |
+
pipe = SpectrogramDiffusionPipeline.from_pretrained(
|
| 9 |
+
"google/music-spectrogram-diffusion", torch_dtype=torch.float16
|
| 10 |
+
).to("cuda")
|
| 11 |
pipe.enable_xformers_memory_efficient_attention()
|
| 12 |
|
| 13 |
processor = MidiProcessor()
|
|
|
|
| 43 |
],
|
| 44 |
title=title,
|
| 45 |
description=description,
|
| 46 |
+
theme="gradio/monochrome",
|
| 47 |
examples=examples,
|
| 48 |
+
).launch(debug=True)
|