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 @@ -206,7 +206,7 @@ ModelState::ModelState(TRITONBACKEND_Model* triton_model)
206
206
TRITONSERVER_Error*
207
207
ModelState::LoadModel (
208
208
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,
210
210
std::shared_ptr<torch::jit::script::Module>* torch_model)
211
211
{
212
212
// Find the TorchScript file that describes the model. If the model
@@ -259,7 +259,7 @@ ModelState::LoadModel(
259
259
if (kind == TRITONSERVER_INSTANCEGROUPKIND_MODEL) {
260
260
// Don't select the device when loading the model.
261
261
torch_model->reset (
262
- new torch::jit::Module (torch::jit::load (model_stream)));
262
+ new torch::jit::Module (torch::jit::load (model_stream)));
263
263
} else {
264
264
torch_model->reset (
265
265
new torch::jit::Module (torch::jit::load (model_stream, device)));
You can’t perform that action at this time.
0 commit comments