Update demos/musicgen_app.py
Browse files- demos/musicgen_app.py +2 -2
demos/musicgen_app.py
CHANGED
@@ -14,7 +14,7 @@ from tempfile import NamedTemporaryFile, gettempdir
|
|
14 |
from einops import rearrange
|
15 |
import torch
|
16 |
import gradio as gr
|
17 |
-
from audiocraft.data.audio_utils import convert_audio
|
18 |
from audiocraft.data.audio import audio_write
|
19 |
from audiocraft.models.encodec import InterleaveStereoCompressionModel
|
20 |
from audiocraft.models import MusicGen, MultiBandDiffusion
|
@@ -201,7 +201,7 @@ def model_worker(model_name: str, task_queue: mp.Queue, result_queue: mp.Queue):
|
|
201 |
if melody_tensor.ndim == 1:
|
202 |
melody_tensor = melody_tensor.unsqueeze(0)
|
203 |
melody_tensor = melody_tensor[..., :int(sr * duration)]
|
204 |
-
processed_melody = convert_audio(melody_tensor, sr, target_sr, target_ac)
|
205 |
if processed_melody is not None:
|
206 |
output, tokens = model.generate_with_chroma(
|
207 |
descriptions=[text],
|
|
|
14 |
from einops import rearrange
|
15 |
import torch
|
16 |
import gradio as gr
|
17 |
+
#from audiocraft.data.audio_utils import convert_audio
|
18 |
from audiocraft.data.audio import audio_write
|
19 |
from audiocraft.models.encodec import InterleaveStereoCompressionModel
|
20 |
from audiocraft.models import MusicGen, MultiBandDiffusion
|
|
|
201 |
if melody_tensor.ndim == 1:
|
202 |
melody_tensor = melody_tensor.unsqueeze(0)
|
203 |
melody_tensor = melody_tensor[..., :int(sr * duration)]
|
204 |
+
#processed_melody = convert_audio(melody_tensor, sr, target_sr, target_ac)
|
205 |
if processed_melody is not None:
|
206 |
output, tokens = model.generate_with_chroma(
|
207 |
descriptions=[text],
|