PLatonG commited on
Commit
d246a8a
·
verified ·
1 Parent(s): 38c5df4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -61,8 +61,8 @@ def evaluate(
61
  stream_output=False,
62
  ):
63
  temperature=0.1
64
- top_p=0.1
65
- top_k=2
66
  num_beams=1
67
  max_new_tokens=380
68
 
@@ -129,6 +129,7 @@ def evaluate(
129
  return_dict_in_generate=True,
130
  output_scores=True,
131
  max_new_tokens=max_new_tokens,
 
132
  )
133
  s = generation_output.sequences[0]
134
  output = tokenizer.decode(s)
 
61
  stream_output=False,
62
  ):
63
  temperature=0.1
64
+ top_p=0.9
65
+ top_k=10
66
  num_beams=1
67
  max_new_tokens=380
68
 
 
129
  return_dict_in_generate=True,
130
  output_scores=True,
131
  max_new_tokens=max_new_tokens,
132
+ seed = 42,
133
  )
134
  s = generation_output.sequences[0]
135
  output = tokenizer.decode(s)