KingNish commited on
Commit
b7dd684
·
verified ·
1 Parent(s): d6882b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -240,7 +240,7 @@ def generate_music(
240
  use_cache=True,
241
  top_k=50,
242
  num_beams=1,
243
- assistant_model=assistant_model
244
  )
245
  if output_seq[0][-1].item() != mmtokenizer.eoa:
246
  tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
 
240
  use_cache=True,
241
  top_k=50,
242
  num_beams=1,
243
+ # assistant_model=assistant_model
244
  )
245
  if output_seq[0][-1].item() != mmtokenizer.eoa:
246
  tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)