KingNish commited on
Commit
39d0c4b
·
verified ·
1 Parent(s): 7245d52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=3
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