提交 6fcf6795 编写于 作者: M MRXLT

remove set_local_bin

上级 61a1c7cb
......@@ -109,9 +109,10 @@ class Server(object):
def set_memory_optimize(self, flag=False):
self.memory_optimization = flag
def set_local_bin(self, path):
def check_local_bin(self):
if "SERVING_BIN" in os.environ:
self.use_local_bin = True
self.bin_path = path
self.bin_path = os.environ["SERVING_BIN"]
def _prepare_engine(self, model_config_path, device):
if self.model_toolkit_conf == None:
......@@ -258,10 +259,11 @@ class Server(object):
def run_server(self):
# just run server with system command
# currently we do not load cube
self.check_local_bin()
if not self.use_local_bin:
self.download_bin()
else:
print("Use local bin")
print("Use local bin : {}".format(self.bin_path))
command = "{} " \
"-enable_model_toolkit " \
"-inferservice_path {} " \
......
......@@ -109,9 +109,10 @@ class Server(object):
def set_memory_optimize(self, flag=False):
self.memory_optimization = flag
def set_local_bin(self, path):
def check_local_bin(self):
if "SERVING_BIN" in os.environ:
self.use_local_bin = True
self.bin_path = path
self.bin_path = os.environ["SERVING_BIN"]
def set_gpuid(self, gpuid=0):
self.gpuid = gpuid
......@@ -243,8 +244,11 @@ class Server(object):
def run_server(self):
# just run server with system command
# currently we do not load cube
self.check_local_bin()
if not self.use_local_bin:
self.download_bin()
else:
print("Use local bin : {}".format(self.bin_path))
command = "{} " \
"-enable_model_toolkit " \
"-inferservice_path {} " \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册