diff --git a/python/paddle_serving_server/__init__.py b/python/paddle_serving_server/__init__.py index e0164b496a458721e0e484b5c6bfa8d139eb5fdb..c399a80d59c58292f8914c32a3faadb76139477f 100644 --- a/python/paddle_serving_server/__init__.py +++ b/python/paddle_serving_server/__init__.py @@ -239,7 +239,7 @@ class Server(object): # it from workflow_conf. default_engine_names = [ 'general_infer_0', 'general_dist_kv_infer_0', - 'general_dist_kv_quant_infer' + 'general_dist_kv_quant_infer_0' ] engine_name = None for node in self.workflow_conf.workflows[0].nodes: diff --git a/python/paddle_serving_server_gpu/__init__.py b/python/paddle_serving_server_gpu/__init__.py index 981494e1a6281780cce633d755034e602dac39c1..db5415127a0d1ef9bcccbd2d0639c3e07083d375 100644 --- a/python/paddle_serving_server_gpu/__init__.py +++ b/python/paddle_serving_server_gpu/__init__.py @@ -274,7 +274,7 @@ class Server(object): # it from workflow_conf. default_engine_names = [ 'general_infer_0', 'general_dist_kv_infer_0', - 'general_dist_kv_quant_infer' + 'general_dist_kv_quant_infer_0' ] engine_name = None for node in self.workflow_conf.nodes: