atwine commited on
Commit
f65cf9a
·
1 Parent(s): e861581

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -27,10 +27,10 @@ class EndpointHandler:
27
  parameters['max_length'] = 1012 # Set your desired max_length here
28
  parameters['min_length'] = 100
29
  parameters['length_penalty'] = 2.0
30
- parameters['num_beams'] = 10
31
  parameters['early_stopping'] = True
32
  parameters['temperature'] = 0.0
33
- parameters['top_k'] = 15
34
  parameters['top_p'] = 1.0
35
 
36
 
 
27
  parameters['max_length'] = 1012 # Set your desired max_length here
28
  parameters['min_length'] = 100
29
  parameters['length_penalty'] = 2.0
30
+ parameters['num_beams'] = 25
31
  parameters['early_stopping'] = True
32
  parameters['temperature'] = 0.0
33
+ parameters['top_k'] = 25
34
  parameters['top_p'] = 1.0
35
 
36