Skip to content

Commit bf76f9c

Browse files
committed
Replace default Triton server uri
1 parent 015a53c commit bf76f9c

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/api_trt/modules/model_zoo/exec_backends/triton_backend.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def parse_model_grpc(model_metadata, model_config):
7474

7575
class Arcface:
7676

77-
def __init__(self, rec_name = 'arcface_r100_v1', model_version='1', triton_uri='10.0.15.109:8002', **kwargs):
77+
def __init__(self, rec_name = 'arcface_r100_v1', model_version='1', triton_uri='localhost:8001', **kwargs):
7878
self.model_name = rec_name
7979
self.model_version = model_version
8080
self.url = triton_uri
@@ -135,7 +135,7 @@ def get_embedding(self,face_img):
135135

136136

137137
class Cosface:
138-
def __init__(self, rec_name = 'arcface_r100_v1', triton_uri='10.0.15.109:8002', model_version='1',**kwargs):
138+
def __init__(self, rec_name = 'arcface_r100_v1', triton_uri='localhost:8001', model_version='1',**kwargs):
139139
self.model_name = rec_name
140140
self.model_version = model_version
141141
self.url = triton_uri
@@ -196,7 +196,7 @@ def get_embedding(self, face_img):
196196

197197
class DetectorInfer:
198198

199-
def __init__(self, model = 'retinaface_r50_v1', output_order=None, triton_uri='10.0.15.109:8002', model_version='1',**kwargs):
199+
def __init__(self, model = 'retinaface_r50_v1', output_order=None, triton_uri='localhost:8001', model_version='1',**kwargs):
200200
self.model_name = model
201201
self.model_version = model_version
202202
self.url = triton_uri

0 commit comments

Comments
 (0)