ford442 commited on
Commit
c72eb33
·
verified ·
1 Parent(s): fa5982d

Update demos/musicgen_app.py

Browse files
Files changed (1) hide show
  1. 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],