Pushpahasa commited on
Commit
e863516
·
1 Parent(s): 8a297b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def song_generator(input_prompt):
20
  # set pad_token_id to eos_token_id because OPT does not have a PAD token
21
  model.config.pad_token_id = model.config.eos_token_id
22
  input_ids = tokenizer(input_prompt, return_tensors="pt")
23
- stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=30)])
24
  outputs = model.contrastive_search(
25
  **input_ids, penalty_alpha=0.6, top_k=15, stopping_criteria=stopping_criteria, pad_token_id=tokenizer.eos_token_id,
26
  )
 
20
  # set pad_token_id to eos_token_id because OPT does not have a PAD token
21
  model.config.pad_token_id = model.config.eos_token_id
22
  input_ids = tokenizer(input_prompt, return_tensors="pt")
23
+ stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=300)])
24
  outputs = model.contrastive_search(
25
  **input_ids, penalty_alpha=0.6, top_k=15, stopping_criteria=stopping_criteria, pad_token_id=tokenizer.eos_token_id,
26
  )