Update handler.py
Browse files- 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'] =
|
31 |
parameters['early_stopping'] = True
|
32 |
parameters['temperature'] = 0.0
|
33 |
-
parameters['top_k'] =
|
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 |
|