KingNish commited on
Commit
0f34fab
·
verified ·
1 Parent(s): e67d49d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -219,14 +219,14 @@ def generate_music(
219
  input_ids=input_ids,
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
 
219
  input_ids=input_ids,
220
  max_new_tokens=max_new_tokens,
221
  min_new_tokens=100,
222
+ do_sample=True,
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