atwine
/

translation-en-lug-v8

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

atwine commited on Oct 17, 2023

Commit

bb04e69

•

1 Parent(s): f65cf9a

Update handler.py

Files changed (1) hide show

handler.py +8 -8

handler.py CHANGED Viewed

@@ -24,14 +24,14 @@ class EndpointHandler:
         # Modify parameters to increase max_length
         if parameters is None:
             parameters = {}
-        parameters['max_length'] = 1012  # Set your desired max_length here
-        parameters['min_length'] = 100
-        parameters['length_penalty'] = 2.0
-        parameters['num_beams'] = 25
-        parameters['early_stopping'] = True
-        parameters['temperature'] = 0.0
-        parameters['top_k'] = 25
-        parameters['top_p'] = 1.0

         # Modify parameters to increase max_length
         if parameters is None:
             parameters = {}
+        # parameters['max_length'] = 1012  # Set your desired max_length here
+        # parameters['min_length'] = 100
+        # parameters['length_penalty'] = 2.0
+        # parameters['num_beams'] = 25
+        # parameters['early_stopping'] = True
+        # parameters['temperature'] = 0.0
+        # parameters['top_k'] = 25
+        # parameters['top_p'] = 1.0