Skip to content

Commit 7a6d46d

Browse files
authored
Use the SetModelConfig backend utility (triton-inference-server#172)
1 parent e9e9c07 commit 7a6d46d

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

src/python_be.cc

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1367,14 +1367,7 @@ ModelState::Create(TRITONBACKEND_Model* triton_model, ModelState** state)
13671367
triton_model, &auto_complete_config));
13681368
if (auto_complete_config) {
13691369
RETURN_IF_ERROR((*state)->LaunchAutoCompleteStubProcess());
1370-
triton::common::TritonJson::WriteBuffer buf;
1371-
(*state)->Stub()->AutoCompleteConfig().Write(&buf);
1372-
1373-
TRITONSERVER_Message* message;
1374-
RETURN_IF_ERROR(TRITONSERVER_MessageNewFromSerializedJson(
1375-
&message, buf.Base(), buf.Size()));
1376-
RETURN_IF_ERROR(TRITONBACKEND_ModelSetConfig(
1377-
triton_model, 1 /* config_version */, message));
1370+
RETURN_IF_ERROR((*state)->SetModelConfig());
13781371
(*state)->ModelConfig() = std::move((*state)->Stub()->AutoCompleteConfig());
13791372

13801373
(*state)->Stub()->UpdateHealth();

0 commit comments

Comments
 (0)