Skip to content

Commit 3efa323

Browse files
committed
Format
1 parent c7d5007 commit 3efa323

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/libtorch.cc

100755100644
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ ModelState::ModelState(TRITONBACKEND_Model* triton_model)
206206
TRITONSERVER_Error*
207207
ModelState::LoadModel(
208208
const std::string& artifact_name, const torch::Device device,
209-
std::string* model_path, const TRITONSERVER_InstanceGroupKind& kind,
209+
std::string* model_path, const TRITONSERVER_InstanceGroupKind& kind,
210210
std::shared_ptr<torch::jit::script::Module>* torch_model)
211211
{
212212
// Find the TorchScript file that describes the model. If the model
@@ -259,7 +259,7 @@ ModelState::LoadModel(
259259
if (kind == TRITONSERVER_INSTANCEGROUPKIND_MODEL) {
260260
// Don't select the device when loading the model.
261261
torch_model->reset(
262-
new torch::jit::Module(torch::jit::load(model_stream)));
262+
new torch::jit::Module(torch::jit::load(model_stream)));
263263
} else {
264264
torch_model->reset(
265265
new torch::jit::Module(torch::jit::load(model_stream, device)));

0 commit comments

Comments
 (0)