Update due to change of triton_client

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
1 year ago · 4f2ad553b6
1 changed files with 1 additions and 1 deletions
--- a/benchmark/qps_test.py
+++ b/benchmark/qps_test.py
@ -66,7 +66,7 @@ if args.num and args.pipe:

 if args.triton:
    client = triton_client.Client(url='localhost:8000')
-    out2 = client(data)[0][0][0]
+    out2 = client(data)[0][0]
    print('Triton: OK')

    if numpy.allclose(out1, out2, atol=args.atol):