Update app.py
Browse files
app.py
CHANGED
@@ -220,7 +220,7 @@ def generate_music(
|
|
220 |
logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
|
221 |
guidance_scale=guidance_scale,
|
222 |
use_cache=True,
|
223 |
-
num_beams=
|
224 |
)
|
225 |
if output_seq[0][-1].item() != mmtokenizer.eoa:
|
226 |
tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
|
@@ -359,7 +359,7 @@ with gr.Blocks() as demo:
|
|
359 |
gr.Examples(
|
360 |
examples=[
|
361 |
[
|
362 |
-
"rap piano street tough piercing vocal hip-hop synthesizer clear vocal male",
|
363 |
"""[verse]
|
364 |
Woke up in the morning, sun is shining bright
|
365 |
Chasing all my dreams, gotta get my mind right
|
|
|
220 |
logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
|
221 |
guidance_scale=guidance_scale,
|
222 |
use_cache=True,
|
223 |
+
num_beams=1
|
224 |
)
|
225 |
if output_seq[0][-1].item() != mmtokenizer.eoa:
|
226 |
tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
|
|
|
359 |
gr.Examples(
|
360 |
examples=[
|
361 |
[
|
362 |
+
"rap piano street tough piercing vocal hip-hop synthesizer clear but aggressive vocal male",
|
363 |
"""[verse]
|
364 |
Woke up in the morning, sun is shining bright
|
365 |
Chasing all my dreams, gotta get my mind right
|