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

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -24,11 +24,11 @@ class EndpointHandler:
24
  # Modify parameters to increase max_length
25
  if parameters is None:
26
  parameters = {}
27
- parameters['max_length'] = 512 # 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
 
24
  # Modify parameters to increase max_length
25
  if parameters is None:
26
  parameters = {}
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