sagu7 commited on
Commit
5123b42
·
1 Parent(s): 9b99b25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def evaluate(input_prompt: InputPrompt):
39
  generation_config=generation_config,
40
  return_dict_in_generate=True,
41
  output_scores=True,
42
- max_new_tokens=50
43
  )
44
  for s in generation_output.sequences:
45
  output = tokenizer.decode(s)
 
39
  generation_config=generation_config,
40
  return_dict_in_generate=True,
41
  output_scores=True,
42
+ max_new_tokens=256
43
  )
44
  for s in generation_output.sequences:
45
  output = tokenizer.decode(s)