Skip to content

Commit 05ae43a

Browse files
committed
Format
1 parent 0b86d4f commit 05ae43a

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
@@ -205,7 +205,7 @@ ModelState::ModelState(TRITONBACKEND_Model* triton_model)
205205
TRITONSERVER_Error*
206206
ModelState::LoadModel(
207207
const std::string& artifact_name, const torch::Device device,
208-
std::string* model_path, const TRITONSERVER_InstanceGroupKind& kind,
208+
std::string* model_path, const TRITONSERVER_InstanceGroupKind& kind,
209209
std::shared_ptr<torch::jit::script::Module>* torch_model)
210210
{
211211
// Find the TorchScript file that describes the model. If the model
@@ -258,7 +258,7 @@ ModelState::LoadModel(
258258
if (kind == TRITONSERVER_INSTANCEGROUPKIND_MODEL) {
259259
// Don't select the device when loading the model.
260260
torch_model->reset(
261-
new torch::jit::Module(torch::jit::load(model_stream)));
261+
new torch::jit::Module(torch::jit::load(model_stream)));
262262
} else {
263263
torch_model->reset(
264264
new torch::jit::Module(torch::jit::load(model_stream, device)));

0 commit comments

Comments
 (0)