diff --git a/paddlenlp/server/handlers/custom_model_handler.py b/paddlenlp/server/handlers/custom_model_handler.py
index c5a133b8ef19..b6b466d5089f 100644
--- a/paddlenlp/server/handlers/custom_model_handler.py
+++ b/paddlenlp/server/handlers/custom_model_handler.py
@@ -26,9 +26,9 @@ def __init__(self):
     def process(cls, predictor, tokenizer, data, parameters):
         max_seq_len = 128
         batch_size = 1
-        if "max_seq_len" not in parameters:
+        if "max_seq_len" in parameters:
             max_seq_len = parameters["max_seq_len"]
-        if "batch_size" not in parameters:
+        if "batch_size" in parameters:
             batch_size = parameters["batch_size"]
         text = None
         if "text" in data:
@@ -97,9 +97,9 @@ def __init__(self):
     def process(cls, predictor, tokenizer, data, parameters):
         max_seq_len = 128
         batch_size = 1
-        if "max_seq_len" not in parameters:
+        if "max_seq_len" in parameters:
             max_seq_len = parameters["max_seq_len"]
-        if "batch_size" not in parameters:
+        if "batch_size" in parameters:
             batch_size = parameters["batch_size"]
         text = None
         if "text" in data: