Saibo commited on
Commit
6ef22f3
·
verified ·
1 Parent(s): 6e0d084

set max_new_tokens to 50 instead of 30

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ if __name__ == "__main__":
43
  grammar_processor = GrammarConstrainedLogitsProcessor(grammar)
44
 
45
  outputs = model.generate(
46
- **inputs, max_new_tokens=30, repetition_penalty=1, return_dict_in_generate=True, output_scores=True, logits_processor=[grammar_processor]
47
  )
48
  # Important: don't forget to set `normalize_logits=True` to obtain normalized probabilities (i.e. sum(p) = 1)
49
  transition_scores = model.compute_transition_scores(outputs.sequences, outputs.scores, normalize_logits=True)
 
43
  grammar_processor = GrammarConstrainedLogitsProcessor(grammar)
44
 
45
  outputs = model.generate(
46
+ **inputs, max_new_tokens=50, repetition_penalty=1, return_dict_in_generate=True, output_scores=True, logits_processor=[grammar_processor]
47
  )
48
  # Important: don't forget to set `normalize_logits=True` to obtain normalized probabilities (i.e. sum(p) = 1)
49
  transition_scores = model.compute_transition_scores(outputs.sequences, outputs.scores, normalize_logits=True)