fix max_length

handler.py

@@ -108,7 +108,7 @@ class EndpointHandler:
         # Check valid inputs
         if not self.check_valid_inputs(input_a, input_b, task):
             return [{"error": "Invalid inputs"}]
-        if "max_new_tokens" not in parameters:
+        if "max_new_tokens" not in parameters and "max_length" not in parameters:
             parameters["max_new_tokens"] = 512
 
         # Tokenize the input