KingNish commited on
Commit
e67d49d
·
verified ·
1 Parent(s): 185e94b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -220,16 +220,16 @@ def generate_music(
220
  max_new_tokens=max_new_tokens,
221
  min_new_tokens=100,
222
  do_sample=False, # just testing what happens
223
- top_p=top_p,
224
- temperature=temperature,
225
- repetition_penalty=repetition_penalty,
226
  eos_token_id=mmtokenizer.eoa,
227
  pad_token_id=mmtokenizer.eoa,
228
  logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
229
- guidance_scale=guidance_scale,
230
  use_cache=True,
231
  # top_k=50,
232
- num_beams=1
233
  )
234
  if output_seq[0][-1].item() != mmtokenizer.eoa:
235
  tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
 
220
  max_new_tokens=max_new_tokens,
221
  min_new_tokens=100,
222
  do_sample=False, # just testing what happens
223
+ # top_p=top_p,
224
+ # temperature=temperature,
225
+ # repetition_penalty=repetition_penalty,
226
  eos_token_id=mmtokenizer.eoa,
227
  pad_token_id=mmtokenizer.eoa,
228
  logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
229
+ # guidance_scale=guidance_scale,
230
  use_cache=True,
231
  # top_k=50,
232
+ # num_beams=1
233
  )
234
  if output_seq[0][-1].item() != mmtokenizer.eoa:
235
  tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)