Update handler.py
Browse files- 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'] =
|
28 |
parameters['min_length'] = 100
|
29 |
parameters['length_penalty'] = 2.0
|
30 |
parameters['num_beams'] = 10
|
31 |
-
|
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
|