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