Replace default Triton server uri

SthPhoenix · Jun 16, 2021 · bf76f9c · bf76f9c
1 parent 015a53c
commit bf76f9c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/api_trt/modules/model_zoo/exec_backends/triton_backend.py b/src/api_trt/modules/model_zoo/exec_backends/triton_backend.py
@@ -74,7 +74,7 @@ def parse_model_grpc(model_metadata, model_config):
 
 class Arcface:
 
-    def __init__(self, rec_name = 'arcface_r100_v1', model_version='1', triton_uri='10.0.15.109:8002', **kwargs):
+    def __init__(self, rec_name = 'arcface_r100_v1', model_version='1', triton_uri='localhost:8001', **kwargs):
         self.model_name = rec_name
         self.model_version = model_version
         self.url = triton_uri
@@ -135,7 +135,7 @@ def get_embedding(self,face_img):
 
 
 class Cosface:
-    def __init__(self, rec_name = 'arcface_r100_v1', triton_uri='10.0.15.109:8002', model_version='1',**kwargs):
+    def __init__(self, rec_name = 'arcface_r100_v1', triton_uri='localhost:8001', model_version='1',**kwargs):
         self.model_name = rec_name
         self.model_version = model_version
         self.url = triton_uri
@@ -196,7 +196,7 @@ def get_embedding(self, face_img):
 
 class DetectorInfer:
 
-    def __init__(self, model = 'retinaface_r50_v1', output_order=None, triton_uri='10.0.15.109:8002', model_version='1',**kwargs):
+    def __init__(self, model = 'retinaface_r50_v1', output_order=None, triton_uri='localhost:8001', model_version='1',**kwargs):
         self.model_name = model
         self.model_version = model_version
         self.url = triton_uri