Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
Commit
•
4b67825
1
Parent(s):
c9a53ea
update
Browse files- app.py +12 -4
- requirements.txt +1 -1
app.py
CHANGED
@@ -2,11 +2,11 @@ import gradio as gr
|
|
2 |
import torchaudio
|
3 |
from audiocraft.models import MusicGen
|
4 |
from audiocraft.data.audio import audio_write
|
5 |
-
import spaces
|
6 |
import logging
|
7 |
import os
|
8 |
import uuid
|
9 |
-
from torch.cuda.amp import autocast
|
10 |
import torch
|
11 |
|
12 |
ZERO_GPU_PATCH_TORCH_DEVICE = 1
|
@@ -18,7 +18,15 @@ logging.info("Carregando o modelo pré-treinado.")
|
|
18 |
model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
|
19 |
model.set_generation_params(duration=30)
|
20 |
|
21 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
def generate_music(description, melody_audio):
|
23 |
with autocast():
|
24 |
logging.info("Iniciando a geração de música.")
|
@@ -73,4 +81,4 @@ gr.Interface(
|
|
73 |
["trap, synthesizer, songstarters, dark, G# minor, 140 bpm", "./assets/kalhonaho.mp3"],
|
74 |
["upbeat, electronic, synth, dance, 120 bpm", None]
|
75 |
]
|
76 |
-
).launch()
|
|
|
2 |
import torchaudio
|
3 |
from audiocraft.models import MusicGen
|
4 |
from audiocraft.data.audio import audio_write
|
5 |
+
#import spaces
|
6 |
import logging
|
7 |
import os
|
8 |
import uuid
|
9 |
+
#from torch.cuda.amp import autocast
|
10 |
import torch
|
11 |
|
12 |
ZERO_GPU_PATCH_TORCH_DEVICE = 1
|
|
|
18 |
model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
|
19 |
model.set_generation_params(duration=30)
|
20 |
|
21 |
+
if torch.cuda.is_available():
|
22 |
+
autocast = torch.cuda.amp.autocast
|
23 |
+
elif torch.backends.mps.is_available():
|
24 |
+
autocast = torch.amp.autocast
|
25 |
+
else:
|
26 |
+
autocast = torch.autocast
|
27 |
+
|
28 |
+
|
29 |
+
#@spaces.GPU(duration=120)
|
30 |
def generate_music(description, melody_audio):
|
31 |
with autocast():
|
32 |
logging.info("Iniciando a geração de música.")
|
|
|
81 |
["trap, synthesizer, songstarters, dark, G# minor, 140 bpm", "./assets/kalhonaho.mp3"],
|
82 |
["upbeat, electronic, synth, dance, 120 bpm", None]
|
83 |
]
|
84 |
+
).launch()
|
requirements.txt
CHANGED
@@ -1,2 +1,2 @@
|
|
1 |
git+https://github.com/facebookresearch/audiocraft#egg=audiocraft
|
2 |
-
torchaudio
|
|
|
1 |
git+https://github.com/facebookresearch/audiocraft#egg=audiocraft
|
2 |
+
#torchaudio
|