From 27610f711f307bf7e604386efee6871e5ab95bfa Mon Sep 17 00:00:00 2001 From: barrierye Date: Wed, 15 Apr 2020 19:34:14 +0800 Subject: [PATCH] fix cube infer op && test=serving --- python/paddle_serving_server/__init__.py | 2 +- python/paddle_serving_server_gpu/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle_serving_server/__init__.py b/python/paddle_serving_server/__init__.py index e0164b49..c399a80d 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 981494e1..db541512 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: -- GitLab