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

fix gpu code

上级 431790cb
......@@ -277,7 +277,7 @@ class Server(object):
'general_dist_kv_quant_infer_0'
]
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:
engine_name = node.name
break
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册