Skip to content

Commit 883df35

Browse files
committed
Format
1 parent 78a66ad commit 883df35

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
@@ -203,7 +203,7 @@ ModelState::ModelState(TRITONBACKEND_Model* triton_model)
203203
TRITONSERVER_Error*
204204
ModelState::LoadModel(
205205
const std::string& artifact_name, const torch::Device device,
206-
std::string* model_path, const TRITONSERVER_InstanceGroupKind& kind,
206+
std::string* model_path, const TRITONSERVER_InstanceGroupKind& kind,
207207
std::shared_ptr<torch::jit::script::Module>* torch_model)
208208
{
209209
// Find the TorchScript file that describes the model. If the model
@@ -256,7 +256,7 @@ ModelState::LoadModel(
256256
if (kind == TRITONSERVER_INSTANCEGROUPKIND_MODEL) {
257257
// Don't select the device when loading the model.
258258
torch_model->reset(
259-
new torch::jit::Module(torch::jit::load(model_stream)));
259+
new torch::jit::Module(torch::jit::load(model_stream)));
260260
} else {
261261
torch_model->reset(
262262
new torch::jit::Module(torch::jit::load(model_stream, device)));

0 commit comments

Comments
 (0)