KingNish commited on
Commit
e47e5a0
·
verified ·
1 Parent(s): 453b42e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -221,7 +221,7 @@ def generate_music(
221
  logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
222
  guidance_scale=guidance_scale,
223
  use_cache=True,
224
- num_beams = 3 if i > 1 else 1
225
  )
226
  if output_seq[0][-1].item() != mmtokenizer.eoa:
227
  tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
 
221
  logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
222
  guidance_scale=guidance_scale,
223
  use_cache=True,
224
+ num_beams = 1
225
  )
226
  if output_seq[0][-1].item() != mmtokenizer.eoa:
227
  tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)