add a new api for models/bert

sunhailin-Leo · Jun 5, 2023 · 3dfdbbf · 3dfdbbf
1 parent 7e05e55
commit 3dfdbbf
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/models/bert/model.go b/models/bert/model.go
@@ -123,6 +123,14 @@ func (m *ModelService) SetModelName(modelPrefix, modelName string) *ModelService
 // GetModelName Get model name.
 func (m *ModelService) GetModelName() string { return m.modelName }
 
+// SetSecondaryServerURL set secondary server url【Only HTTP】
+func (m *ModelService) SetSecondaryServerURL(url string) *ModelService {
+	if m.tritonService != nil {
+		m.tritonService.SetSecondaryServerURL(url)
+	}
+	return m
+}
+
 ////////////////////////////////////////////////// Flag Switch API //////////////////////////////////////////////////
 
 ///////////////////////////////////////// Bert Service Pre-Process Function /////////////////////////////////////////