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