提交 61f07b82 编写于 作者: B barrierye

fix gpu code

上级 431790cb
...@@ -277,7 +277,7 @@ class Server(object): ...@@ -277,7 +277,7 @@ class Server(object):
'general_dist_kv_quant_infer_0' 'general_dist_kv_quant_infer_0'
] ]
engine_name = None engine_name = None
for node in self.workflow_conf.nodes: for node in self.workflow_conf.workflows[0].nodes:
if node.name in default_engine_names: if node.name in default_engine_names:
engine_name = node.name engine_name = node.name
break break
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册