diff --git a/python/paddle_serving_server/__init__.py b/python/paddle_serving_server/__init__.py index cf7b1126267be047bcc41232e5ba03b78b51d758..1e5fd16ed6c153a28cd72422ca3ef7b9177cb079 100644 --- a/python/paddle_serving_server/__init__.py +++ b/python/paddle_serving_server/__init__.py @@ -655,7 +655,11 @@ class MultiLangServer(object): server_config_paths) self.bclient_config_path_ = client_config_path - def prepare_server(self, workdir=None, port=9292, device="cpu"): + def prepare_server(self, + workdir=None, + port=9292, + device="cpu", + cube_conf=None): if not self._port_is_available(port): raise SystemExit("Prot {} is already used".format(port)) default_port = 12000 @@ -666,7 +670,10 @@ class MultiLangServer(object): self.port_list_.append(default_port + i) break self.bserver_.prepare_server( - workdir=workdir, port=self.port_list_[0], device=device) + workdir=workdir, + port=self.port_list_[0], + device=device, + cube_conf=cube_conf) self.set_port(port) def _launch_brpc_service(self, bserver): diff --git a/python/paddle_serving_server_gpu/__init__.py b/python/paddle_serving_server_gpu/__init__.py index d4244ea4b42ed56af360aca7bce37bfb3029078c..df04cb7840bbacd90ccb7e3c66147a6856b23e02 100644 --- a/python/paddle_serving_server_gpu/__init__.py +++ b/python/paddle_serving_server_gpu/__init__.py @@ -699,7 +699,11 @@ class MultiLangServer(object): server_config_paths) self.bclient_config_path_ = client_config_path - def prepare_server(self, workdir=None, port=9292, device="cpu"): + def prepare_server(self, + workdir=None, + port=9292, + device="cpu", + cube_conf=None): if not self._port_is_available(port): raise SystemExit("Prot {} is already used".format(port)) default_port = 12000 @@ -710,7 +714,10 @@ class MultiLangServer(object): self.port_list_.append(default_port + i) break self.bserver_.prepare_server( - workdir=workdir, port=self.port_list_[0], device=device) + workdir=workdir, + port=self.port_list_[0], + device=device, + cube_conf=cube_conf) self.set_port(port) def _launch_brpc_service(self, bserver):