File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -203,7 +203,7 @@ ModelState::ModelState(TRITONBACKEND_Model* triton_model)
203
203
TRITONSERVER_Error*
204
204
ModelState::LoadModel (
205
205
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,
207
207
std::shared_ptr<torch::jit::script::Module>* torch_model)
208
208
{
209
209
// Find the TorchScript file that describes the model. If the model
@@ -256,7 +256,7 @@ ModelState::LoadModel(
256
256
if (kind == TRITONSERVER_INSTANCEGROUPKIND_MODEL) {
257
257
// Don't select the device when loading the model.
258
258
torch_model->reset (
259
- new torch::jit::Module (torch::jit::load (model_stream)));
259
+ new torch::jit::Module (torch::jit::load (model_stream)));
260
260
} else {
261
261
torch_model->reset (
262
262
new torch::jit::Module (torch::jit::load (model_stream, device)));
You can’t perform that action at this time.
0 commit comments