Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,6 +34,7 @@ JS = """function () {
|
|
| 34 |
if torch.cuda.is_available():
|
| 35 |
pipe = StableAudioPipeline.from_pretrained(
|
| 36 |
model,
|
|
|
|
| 37 |
torch_dtype=torch.float16).to("cuda")
|
| 38 |
|
| 39 |
|
|
@@ -69,7 +70,7 @@ def generate_image(
|
|
| 69 |
|
| 70 |
# Gradio Interface
|
| 71 |
|
| 72 |
-
with gr.Blocks(theme='soft', css=
|
| 73 |
with gr.Accordion(""):
|
| 74 |
gr.Markdown(DESCRIPTION)
|
| 75 |
with gr.Row():
|
|
|
|
| 34 |
if torch.cuda.is_available():
|
| 35 |
pipe = StableAudioPipeline.from_pretrained(
|
| 36 |
model,
|
| 37 |
+
low_cpu_mem_usage=True,
|
| 38 |
torch_dtype=torch.float16).to("cuda")
|
| 39 |
|
| 40 |
|
|
|
|
| 70 |
|
| 71 |
# Gradio Interface
|
| 72 |
|
| 73 |
+
with gr.Blocks(theme='soft', css=CSS, title="Stable Audio Open") as iface:
|
| 74 |
with gr.Accordion(""):
|
| 75 |
gr.Markdown(DESCRIPTION)
|
| 76 |
with gr.Row():
|