vpcom commited on
Commit
38c2ed3
1 Parent(s): fa8e124

feat: return output_scores as well

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -575,6 +575,7 @@ def generate(
575
  top_p = float(top_p)
576
  generate_kwargs = dict(
577
  max_new_tokens=max_new_tokens,
 
578
  temperature=None if temperature==1 else temperature,
579
  top_p=None if top_p==0 else top_p,
580
  repetition_penalty=None if repetition_penalty==1 else repetition_penalty,
 
575
  top_p = float(top_p)
576
  generate_kwargs = dict(
577
  max_new_tokens=max_new_tokens,
578
+ output_scores=True,
579
  temperature=None if temperature==1 else temperature,
580
  top_p=None if top_p==0 else top_p,
581
  repetition_penalty=None if repetition_penalty==1 else repetition_penalty,